search

aditya-grover / climate-learn

Source code for ClimateLearn
MIT License
310 stars 50 forks source link

Current ViT implementation works with timm 0.6.12 and not with 0.9.2 #102

Closed prakhar6sharma closed 1 year ago

prakhar6sharma commented 1 year ago

Describe the bug Running pytest tests/ gives an error for test_vit() when running with the version 0.9.2 but works perfectly with version 0.6.12.

To Reproduce Create a fresh conda environment with python 3.7 and install all the dependencies.

Screenshots Screenshot 2023-05-15 at 5 51 55 AM

jasonjewik commented 1 year ago

I was able to reproduce this bug. I need to re-assess ViT anyway to make it conform with the rest of the updated models, so I'll update it to use the latest version of timm and then peg timm to version 0.9.2 in pyproject.toml.

prakhar6sharma commented 1 year ago

@jasonjewik, I have PRs (#101 and #103) failing in their build because of this issue.