图像分类的例子如何单机多卡进行训练
Created by: Sampson1107
请问在如下目录下,执行修改train.py的如下行:
paddle.init(use_gpu=True, trainer_count=1)为
paddle.init(use_gpu=True, trainer_count=4)
是否可以在四个卡上数据并行训练? 若可以,我设置的batch_size=1024,是四个卡总的batch_size,还是单卡的batch_size呢? 在这个目录下我仅仅修改了如下注释行(训练和测试部分):
flowers.train()
为
reader.train_reader('train.list')
和num_classes实现了在imagenet数据集上的训练,是否可行? 期待您的回复。