Minor change to work in pytorch 1.8.1 about savn HOT 6 CLOSED

Hi,

Do you succeed to train their models?

I tried to train their SAVN model, but I might have been stuck in the environments settings (e.g. python, torch, torchvision versions). I always met some errors when I ran their codes for training.

from savn.

Yes, I was able to run training and evaluation scripts for savn. But I didn't end up using this method as a baseline. I found it difficult to get this to work on the newer version of Ai2thor (3.3.4), because this codebase requires precomputed resnet features. The existing trained model (trained on ai2thor 1.0.0) doesn't work well with images from the new version.

from savn.

Would you mind giving me the package versions you use to train their models? (maybe you can run conda list or pip list to obtain the information)

It is impossible to install torch==0.4.1 now, and I install PyTorch using: conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=10.2 -c pytorch. However I met some errors.

from savn.

This is I believe the state of pytorch packages when I was working on this:

$ pip list | grep torch
efficientnet-pytorch        0.6.3
pytorch-lightning           1.3.8
segmentation-models-pytorch 0.2.0
torch                       1.9.0+cu111
torchaudio                  0.9.0
torchmetrics                0.4.1
torchvision                 0.10.0+cu111

from savn.

Thanks for your information! I have run their codes successfully now.

from savn.

@YunlianMoon Glad it helped!

from savn.