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

latest params and relative paths

Browse Source
This commit is contained in:
Si11ium
2020-05-21 12:14:03 +02:00
parent 444725f6af
commit e04ef2f8b9
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
main.py
View File

@ -7,10 +7,10 @@ import torch
from pytorch_lightning import Trainer
from pytorch_lightning.callbacks import ModelCheckpoint, EarlyStopping
from modules.utils import LightningBaseModule
from utils.config import Config
from utils.logging import Logger
from utils.model_io import SavedLightningModels
from ml_lib.modules.util import LightningBaseModule
from ml_lib.utils.config import Config
from ml_lib.utils.logging import Logger
from ml_lib.utils.model_io import SavedLightningModels
warnings.filterwarnings('ignore', category=FutureWarning)
warnings.filterwarnings('ignore', category=UserWarning)
@ -76,6 +76,6 @@ def run_lightning_loop(config_obj):
if __name__ == "__main__":
from _templates.new_project._parameters import args
from ._parameters import args
config = Config.read_namespace(args)
trained_model = run_lightning_loop(config)