If you want to use a smaller number of GPUs, you need to modify .yaml configuration files in configs/. Specifically, you need to modify the NUM_GPUS, TRAIN.BATCH_SIZE, TEST.BATCH_SIZE, DATA_LOADER.NUM_WORKERS entries in each configuration file. The BATCH_SIZE entry should be the same or higher as the NUM_GPUS entry. In configs/Kinetics/TimeSformer_divST_8x32_224_4gpus.yaml, we provide a sample configuration file for a 4 GPU setu的含义
时间: 2024-01-24 14:18:46 浏览: 385
Android 性能优化一: 9.Smaller PNG Files (100 Days of Google Dev).mp4
这段话的意思是,如果你想使用更少的GPU训练模型,你需要修改`configs/`目录下的`.yaml`配置文件。具体来说,你需要修改每个配置文件中的`NUM_GPUS`、`TRAIN.BATCH_SIZE`、`TEST.BATCH_SIZE`、`DATA_LOADER.NUM_WORKERS`条目。`BATCH_SIZE`条目应该与`NUM_GPUS`条目相同或更高。在`configs/Kinetics/TimeSformer_divST_8x32_224_4gpus.yaml`中,我们提供了一个4 GPU设置的示例配置文件。
阅读全文