protsenkovi 6 months ago
parent e5a4c9531a
commit 632d1ab7d6

@ -94,10 +94,7 @@ def valid_steps(model, datasets, config, log_prefix=""):
'Total area',
'Total time, s'
]
config.logger.info(pd.DataFrame(
[row],
columns=column_names
).set_index('Dataset'))
config.logger.info("\n" + str(pd.DataFrame([row], columns=column_names).set_index('Dataset').T))
config.writer.flush()
results = pd.DataFrame(results, columns=column_names).set_index('Dataset')

@ -50,14 +50,15 @@ class TrainOptions:
def parse_args(self):
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.train_datasets = args.train_datasets.split(',')
args.val_datasets = args.val_datasets.split(',')
return args
def __repr__(self):
config = self.parser.parse_args()
config = self.parse_args()
message = ''
message += '----------------- Options ---------------\n'
for k, v in sorted(vars(config).items()):

Loading…
Cancel
Save