crosstyan/OpenGait
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix error message

Browse Source
This commit is contained in:
darkliang
2022-03-18 21:14:56 +08:00
parent 538c40ba87
commit c5919708ab
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/main.py
+1 -1
View File
@@ -58,7 +58,7 @@ if __name__ == '__main__':
torch.distributed.init_process_group('nccl', init_method='env://') torch.distributed.init_process_group('nccl', init_method='env://')
if torch.distributed.get_world_size() != torch.cuda.device_count(): if torch.distributed.get_world_size() != torch.cuda.device_count():
raise ValueError("Expect number of availuable GPUs({}) equals to the world size({}).".format( raise ValueError("Expect number of availuable GPUs({}) equals to the world size({}).".format(
torch.distributed.get_world_size(), torch.cuda.device_count())) torch.cuda.device_count(), torch.distributed.get_world_size()))
cfgs = config_loader(opt.cfgs) cfgs = config_loader(opt.cfgs)
if opt.iter != 0: if opt.iter != 0:
cfgs['evaluator_cfg']['restore_hint'] = int(opt.iter) cfgs['evaluator_cfg']['restore_hint'] = int(opt.iter)