|
|
@ -50,14 +50,15 @@ class TrainOptions:
|
|
|
|
|
|
|
|
|
|
|
|
def parse_args(self):
|
|
|
|
def parse_args(self):
|
|
|
|
args = self.parser.parse_args()
|
|
|
|
args = self.parser.parse_args()
|
|
|
|
args.models_dir = Path(args.models_dir)
|
|
|
|
args.datasets_dir = Path(args.datasets_dir).resolve()
|
|
|
|
|
|
|
|
args.models_dir = Path(args.models_dir).resolve()
|
|
|
|
args.model_path = Path(args.model_path) if not args.model_path is None else None
|
|
|
|
args.model_path = Path(args.model_path) if not args.model_path is None else None
|
|
|
|
args.train_datasets = args.train_datasets.split(',')
|
|
|
|
args.train_datasets = args.train_datasets.split(',')
|
|
|
|
args.val_datasets = args.val_datasets.split(',')
|
|
|
|
args.val_datasets = args.val_datasets.split(',')
|
|
|
|
return args
|
|
|
|
return args
|
|
|
|
|
|
|
|
|
|
|
|
def __repr__(self):
|
|
|
|
def __repr__(self):
|
|
|
|
config = self.parser.parse_args()
|
|
|
|
config = self.parse_args()
|
|
|
|
message = ''
|
|
|
|
message = ''
|
|
|
|
message += '----------------- Options ---------------\n'
|
|
|
|
message += '----------------- Options ---------------\n'
|
|
|
|
for k, v in sorted(vars(config).items()):
|
|
|
|
for k, v in sorted(vars(config).items()):
|
|
|
|