search

omihub777 / ViT-CIFAR

PyTorch implementation for Vision Transformer[Dosovitskiy, A.(ICLR'21)] modified to obtain over 90% accuracy FROM SCRATCH on CIFAR-10 with small number of parameters (= 6.3M, originally ViT-B has 86M).
MIT License
167 stars 26 forks source link

AttributeError: Can't pickle local object 'SubPolicy.__init__.<locals>.<lambda>' #7

Closed SJTUzhou closed 7 months ago

SJTUzhou commented 1 year ago

Same issue, please see https://github.com/DeepVoltaire/AutoAugment/issues/16. Solution found at https://github.com/DeepVoltaire/AutoAugment/pull/30/commits Please update autoaugment.py and ops.py

omihub777 commented 7 months ago

@SJTUzhou I'm sorry for not getting back to you sooner. I applied a fix patch according to your reference just now. Thank you!