0
点赞
收藏
分享

微信扫一扫

报错: AttributeError: Can‘t pickle local object ‘trainer_synapse.<locals>.worker_init_fn‘

AttributeError: Can’t pickle local object ‘trainer_synapse..worker_init_fn’

解决方案:
DataLoader的参数num_workers设置为0。

    trainloader = DataLoader(db_train, batch_size=batch_size, shuffle=True, num_workers=0, pin_memory=True,
                             worker_init_fn=worker_init_fn)
举报

相关推荐

0 条评论