def __init__(self, args):
sub_dir = datetime.strftime(datetime.now(),
'%m%d-%H%M%S') # prepare saving path
self.save_dir = os.path.join(args.save_dir, sub_dir)
if not os.path.exists(self.save_dir):
os.makedirs(self.save_dir)
setlogger(os.path.join(self.save_dir, 'train.log')) # set logger
for k, v in args.__dict__.items(): # save args
logging.info("{}: {}".format(k, v))
self.args = args
parser.add_argument('--print_step',
type=int,
default=50,
help='the interval of log training information')
args = parser.parse_args()
return args
if __name__ == '__main__':
args = parse_args()
os.environ['CUDA_VISIBLE_DEVICES'] = args.cuda_device.strip()
# Prepare the saving path for the model
sub_dir = args.model_name + '_' + datetime.strftime(
datetime.now(), '%m%d-%H%M%S')
save_dir = os.path.join(args.checkpoint_dir, sub_dir)
if not os.path.exists(save_dir):
os.makedirs(save_dir)
# set the logger
setlogger(os.path.join(save_dir, 'train.log'))
# save the args
for k, v in args.__dict__.items():
logging.info("{}: {}".format(k, v))
trainer = train_utils(args, save_dir)
trainer.setup()
trainer.train()
if __name__ == '__main__':
start_tm = datetime.now()
args = parse_args()
os.environ['CUDA_VISIBLE_DEVICES'] = args.cuda_device.strip()
# Prepare the saving path for the test results
sub_dir = args.model_name+'_'+args.data_name + '_' + datetime.strftime(start_tm, '%m%d-%H%M%S')
save_dir = os.path.join(args.results_dir, sub_dir)
if not os.path.exists(save_dir):
os.makedirs(save_dir)
# Calculate the model file name
model_file_name = model_file_name_from_args(args)
# set the logger
setlogger(os.path.join(save_dir, 'testing.log'))
# save the args
for k, v in args.__dict__.items():
logging.info("{}: {}".format(k, v))
tester = test_utils(args, save_dir, model_file_name)
tester.setup()
results = tester.test()
end_tm = datetime.now()
duration = end_tm - start_tm
report(args=args, results=results, duration=duration, save_dir=save_dir)
help='batchsize of the training process')
parser.add_argument('--num_workers',
type=int,
default=0,
help='the number of training process')
args = parser.parse_args()
return args
if __name__ == '__main__':
args = parse_args()
os.environ['CUDA_VISIBLE_DEVICES'] = args.cuda_device.strip()
# get model name
sub_dir = args.checkpoint_dir.split("\\")[-1]
model_name = args.checkpoint_subdir.split("_")[0]
# get the checkpoint file of best model
path = args.checkpoint_dir
f_list = os.listdir(path)
files = [i for i in f_list if '.pth' in i]
files.sort(key=lambda x: float(x.split('-')[1]), reverse=True)
checkpoint_file = os.path.join(path, files[0])
# set the test logger
setlogger(os.path.join(path, 'test.log'))
evaluater = Evaluate_Utils(args, model_name, checkpoint_file)
evaluater.evaluate()