2023年3月7日
DP训练的效率始终提不上去,不知道怎么解决,太慢了
问题1:AttributeError: ‘DataParallel’ object has no attribute ‘generator’
找不到模型的某些对象,就是因为DP对象多封装了一层“.module”
解决:加上“.module”
train_model.module.generator
2023年3月7日
DP训练的效率始终提不上去,不知道怎么解决,太慢了
问题1:AttributeError: ‘DataParallel’ object has no attribute ‘generator’
找不到模型的某些对象,就是因为DP对象多封装了一层“.module”
解决:加上“.module”
train_model.module.generator