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

Fixed a spelling mistake.

Browse Source
This commit is contained in:
DavidLee528
2023-05-15 16:28:52 +08:00
committed by GitHub
parent 2740b8eece
commit 1784af34a2
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: