OpenGait代码通常通过命令行执行,例如:
CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./configs/gaitset/gaitset.yaml --phase train
如果希望在 PyCharm 中直接运行单卡训练而无需使用命令行,可以按照以下步骤进行设置。
步骤 1: 打开 Run -> Edit Configurations...
:
设置内容如下:
- Script path:
/home/user/LSM/OpenGait/openga