“DP”训练模型发生错误

TrustAIResearch / MLHospital

MIT License

43 stars 6 forks source link

“DP”训练模型发生错误 #2

Open thulx18 opened 1 year ago

thulx18 commented 1 year ago

您好，我在mlh/example目录下，运行"python train_target_models.py --mode target --training_type DP"，发生报错报错信息为“AttributeError: Can't pickle local object 'wrap_collate_with_empty..collate”

xinleihe commented 1 year ago

您好，我们在复现的时候没有发现相关报错，请问您能提供更下详细的报错信息么？同时，请您检查一下opacus的版本，我们目前使用的是1.3.0