parser.add_argument('--device', choices=['gpu', 'cpu']) parser.add_argument('-c', '--config', metavar='C', default=None, help='The Configuration file') args = parser.parse_args() update_cfg_from_cfg(search_cfg, cfg) if args.config is not None: merge_cfg_from_file(args.config, cfg) config = cfg args.save = os.path.join(args.save, 'output') utils.create_exp_dir(args.save) args.input_size = eval(args.input_size) if len(args.input_size) != 4: raise ValueError('The batch size should be specified.') log_format = '%(asctime)s %(message)s' logging.basicConfig(stream=sys.stdout, level=logging.INFO, format=log_format, datefmt='%m/%d %I:%M:%S %p') fh = logging.FileHandler(os.path.join(args.save, 'log.txt')) fh.setFormatter(logging.Formatter(log_format)) logging.getLogger().addHandler(fh) if not torch.cuda.is_available():
parser.add_argument('--loss_scale', type=str, default=None) parser.add_argument('--channels_last', type=bool, default=False) # others parser.add_argument('--seed', type=int, default=2, help='random seed') parser.add_argument('--note', type=str, default='try', help='note for this run') args, unparsed = parser.parse_known_args() args.save = os.path.join( args.save, 'eval-{}-{}'.format(time.strftime("%Y%m%d-%H%M%S"), args.note)) if args.local_rank == 0: create_exp_dir(args.save, scripts_to_save=None) log_format = '%(asctime)s %(message)s' logging.basicConfig(stream=sys.stdout, level=logging.INFO, format=log_format, datefmt='%m/%d %I:%M:%S %p') fh = logging.FileHandler(os.path.join(args.save, 'log.txt')) fh.setFormatter(logging.Formatter(log_format)) logging.getLogger().addHandler(fh) if hasattr(torch, 'channels_last') and hasattr(torch, 'contiguous_format'): if args.channels_last: memory_format = torch.channels_last else: memory_format = torch.contiguous_format