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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
_parameters.py
View File

@ -27,8 +27,8 @@ main_arg_parser.add_argument("--data_additional_resource_root", type=str, defaul
main_arg_parser.add_argument("--data_use_preprocessed", type=strtobool, default=True, help="")
# Transformations
main_arg_parser.add_argument("--transformations_to_tensor", type=strtobool, default=False, help="")
main_arg_parser.add_argument("--transformations_normalize", type=strtobool, default=False, help="")
# main_arg_parser.add_argument("--transformations_to_tensor", type=strtobool, default=False, help="")
# main_arg_parser.add_argument("--transformations_normalize", type=strtobool, default=False, help="")
# Transformations
main_arg_parser.add_argument("--train_outpath", type=str, default="output", help="")

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)