steffen/ml_lib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Code Comments, Getting Dirty Env, Naming

Browse Source
This commit is contained in:
Steffen Illium
2021-05-11 10:31:34 +02:00
parent faa27c3cf9
commit ab01006eae
7 changed files with 51 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
_templates/new_project/main.py
View File

@ -9,7 +9,7 @@ from pytorch_lightning.callbacks import ModelCheckpoint, EarlyStopping
from ml_lib.modules.util import LightningBaseModule
from ml_lib.utils.config import Config
from ml_lib.utils.loggers import Logger
from ml_lib.utils.loggers import LightningLogger
warnings.filterwarnings('ignore', category=FutureWarning)
warnings.filterwarnings('ignore', category=UserWarning)
@ -20,7 +20,7 @@ def run_lightning_loop(config_obj):
# Logging
# ================================================================================
# Logger
with Logger(config_obj) as logger:
with LightningLogger(config_obj) as logger:
# Callbacks
# =============================================================================
# Checkpoint Saving