'network.distributed', default=False):
os.environ["CUDA_VISIBLE_DEVICES"] = ','.join(
str(gpu_id) for gpu_id in configer.get('gpu'))
if configer.get('network', 'norm_type') is None:
configer.update('network.norm_type', 'batchnorm')
if torch.cuda.device_count() <= 1 or configer.get('network.distributed',
default=False):
configer.update('network.gather', True)
project_dir = os.path.dirname(os.path.realpath(__file__))
configer.add('project_dir', project_dir)
Log.init(log_level=configer.get('logging', 'log_level'),
log_format=configer.get('logging', 'log_format'),
distributed_rank=configer.get('local_rank'))
Log.info('BN Type is {}.'.format(configer.get('network', 'norm_type')))
Log.info('Config Dict: {}'.format(json.dumps(configer.to_dict(),
indent=2)))
runner_selector = RunnerSelector(configer)
runner = None
if configer.get('task') == 'pose':
runner = runner_selector.pose_runner()
elif configer.get('task') == 'seg':
runner = runner_selector.seg_runner()
elif configer.get('task') == 'det':
runner = runner_selector.det_runner()
elif configer.get('task') == 'cls':
'network', 'norm_type')
project_dir = os.path.dirname(os.path.realpath(__file__))
configer.add('project_dir', project_dir)
if configer.get('logging', 'log_to_file'):
log_file = configer.get('logging', 'log_file')
new_log_file = '{}_{}'.format(
log_file, time.strftime("%Y-%m-%d_%X", time.localtime()))
configer.update('logging.log_file', new_log_file)
else:
configer.update('logging.logfile_level', None)
Log.init(logfile_level=configer.get('logging', 'logfile_level'),
stdout_level=configer.get('logging', 'stdout_level'),
log_file=configer.get('logging', 'log_file'),
log_format=configer.get('logging', 'log_format'),
rewrite=configer.get('logging', 'rewrite'))
Log.info('BN Type is {}.'.format(configer.get('network', 'norm_type')))
Log.info('Config Dict: {}'.format(json.dumps(configer.to_dict(),
indent=2)))
runner_selector = RunnerSelector(configer)
runner = None
if configer.get('task') == 'pose':
runner = runner_selector.pose_runner()
elif configer.get('task') == 'seg':
runner = runner_selector.seg_runner()
elif configer.get('task') == 'det':
runner = runner_selector.det_runner()
