stereo-transformer
stereo-transformer copied to clipboard
BN param &checkpoint error
Thank you for giving the code. While I have some questions for sh files. My enviroment is python 3.9.19, cuda 11.1, torch 1.9.1.
I tried two ways to run the code.
First is sh the sh file, while it show me the pic as below
I'm not sure is this a problem for pth file or otherwise, did you have the problem?
second I run this in the terminal, python main.py --resume /home/fei/Github/stereo-transformer/kitti_finetuned_model.pth.tar
For this I changed my cuda & Pytorch Version( before CUDA 11.8 Pytorch 2.0.1 ) while it seems the BN para doesn't have relationship with the environment version, so how to deal with this problem?