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

***************************************************

码字不易,收藏之余,别忘了给我点个赞吧!

***************************************************

---------Start

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)

猜你喜欢

转载自blog.csdn.net/qq_37652891/article/details/123460932
今日推荐