search

ucbrise / actnn

ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training
MIT License
196 stars 30 forks source link

AttributeError: 'QModule' object has no attribute 'stride' #33

Closed wemindful closed 2 years ago

wemindful commented 2 years ago

AttributeError: 'QModule' object has no attribute 'stride'

wemindful commented 2 years ago

Traceback (most recent call last): File "train.py", line 624, in main(opt) File "train.py", line 521, in main train(opt.hyp, opt, device, callbacks) File "train.py", line 198, in train gs = max(int(model.stride.max()), 32) # grid size (max stride) File "/usr/local/python3.7/lib/python3.7/site-packages/torch/nn/modules/module.py", line 948, in getattr type(self).name, name)) AttributeError: 'QModule' object has no attribute 'stride'