in train_kinetics.py [0:0]
def autofill(args):
# customized
if not args.task_name:
args.task_name = os.path.basename(os.getcwd())
if not args.log_file:
if os.path.exists("./exps/logs"):
if args.distributed:
with open('Hosts') as f:
content = f.readlines()
rank = int([x for x in range(len(content)) if socket.gethostname() in content[x].lower()][0])
args.log_file = "./exps/logs/{}_rank-{}.log".format(args.task_name, rank)
else:
args.log_file = "./exps/logs/{}.log".format(args.task_name)
else:
args.log_file = ".{}.log".format(args.task_name)
# fixed
args.model_prefix = os.path.join(args.model_dir, args.task_name)
return args