abhishekkrthakur/tez

Atrribute Error while importing tez

AnuraagBeniwal2610 opened this issue · 5 comments

AnuraagBeniwal2610 commented

when i try to import tez using - import tez. I am encountering this error
AttributeError: module 'torch.optim.lr_scheduler' has no attribute 'SAVE_STATE_WARNING'

abhishekkrthakur commented

which version of torch are you using?

AnuraagBeniwal2610 commented

i am using 1.8.0, which was automatically installed when i installed tez

abhishekkrthakur commented

yeah, i might need to fix that. 1.7.1 should work fine. if you could use it, it would be great.

for 1.8, ill make a new release soon.

AnuraagBeniwal2610 commented

Ok sure, thanks for such a fast reply. Loved your book btw as well as your videos.

MrR0b0t-23 commented

I encountered this error
AttributeError: module 'torch.optim.lr_scheduler' has no attribute 'SAVE_STATE_WARNING'

To solve this error I simply removed "warnings.filterwarnings("ignore", message=torch.optim.lr_scheduler.SAVE_STATE_WARNING)" line from tez/tez/model/model.py

After changing this my program worked properly.