search

huawei-noah / vega

AutoML tools chain
http://www.noahlab.com.hk/opensource/vega/
Other
841 stars 177 forks source link

多GPU训练 #258

Open bondruy opened 1 year ago

bondruy commented 1 year ago

请问多GPU训练一个模型应该怎么配置

zhangjiajin commented 1 year ago

请参考这个： https://github.com/huawei-noah/vega/blob/master/examples/features/data_parallel_train/horovod.yml 目前只支持数据并行。