联邦学习pytorch踩坑

创建多进程训练模型报错，要设置子进程创建模式为spwan

在子进程中将模型放到GPU上，不然会报错

posted @ 2023-06-20 23:33  chooise  阅读(16)  评论(0)    收藏  举报