fix BN issue due to torch version #8

mli0603 / stereo-transformer

Revisiting Stereo Depth Estimation From a Sequence-to-Sequence Perspective with Transformers. (ICCV 2021 Oral)

Apache License 2.0

659 stars 107 forks source link

fix BN issue due to torch version #8 #49

Closed DeH40 closed 2 years ago

mli0603 commented 2 years ago

@DeH40

Thanks for the PR! I have verified this works on my end. However, this will introduce problems with model loading because now it expects running_var and running_mean in the state dict. I can fix this on your PR if you want.

Do you mind giving me permission to change this PR? The instruction is here: https://docs.github.com/en/pull-requests/collaborating-with-pull-requests/working-with-forks/allowing-changes-to-a-pull-request-branch-created-from-a-fork

Thanks again!

DeH40 commented 2 years ago

Sure，you can fix the loading problem on my PR 😄 .