diff --git a/opengait/main.py b/opengait/main.py index 35d372c..598dcaa 100644 --- a/opengait/main.py +++ b/opengait/main.py @@ -59,7 +59,7 @@ def run_model(cfgs, training): if __name__ == '__main__': torch.distributed.init_process_group('nccl', init_method='env://') 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())) cfgs = config_loader(opt.cfgs) if opt.iter != 0: