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

Merge pull request #133 from DavidLee528/patch-1

Browse Source 
Fixed a spelling mistake.
This commit is contained in:
Chao Fan
2023-06-06 19:04:43 +08:00
committed by GitHub
parent e5be2028dd 1784af34a2
commit 9ff2520c05
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
opengait/main.py
+1 -1
View File
@@ -59,7 +59,7 @@ def run_model(cfgs, training):
if __name__ == '__main__': 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 available GPUs({}) equals to the world size({}).".format(
torch.cuda.device_count(), torch.distributed.get_world_size())) 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: