search

gasharper / PyramidFlow

[CVPR 2023] PyramidFlow: High-Resolution Defect Contrastive Localization using Pyramid Normalizing Flow
MIT License
56 stars 14 forks source link

您好，train过程遇到的问题 #8

Closed caihaocong closed 1 year ago

caihaocong commented 1 year ago

您好，在train过程中，加载数据集遇到 “AttributeError: Can't pickle local object 'fix_randseed..seed_worker'” 的问题，请问您有遇到过吗？

gasharper commented 1 year ago

我是在linux下训练的，pytorch需要版本>=1.9 可能是windows下多进程跟linux下不一样的缘故。 看看设置num_workers=0是否有用？