Traceback (most recent call last):
File “F:/QLing/QLing.Code/FastFlow/main.py”, line 31, in 
model = train(train_loader, test_loader)
File “F:\QLing\QLing.Code\FastFlow\train.py”, line 146, in train
save_model(model, c.modelname)
File “F:\QLing\QLing.Code\FastFlow\model.py”, line 142, in save_model
torch.save(model, os.path.join(MODEL_DIR, filename))
File “D:\Programs\Python\Python37\lib\site-packages\torch\serialization.py”, line 379, in save
_save(obj, opened_zipfile, pickle_module, pickle_protocol)
File “D:\Programs\Python\Python37\lib\site-packages\torch\serialization.py”, line 484, in _save
pickler.dump(obj)
AttributeError: Can’t pickle local object ‘AllInOneBlock.init..’