For some reason it throws missing keys, but couldn't find source of problem.

iVMCL / AOGNet-v2

Official implementation of AOGNets and Attentive Normalization.

Other

45 stars 8 forks source link

Closed nile649 closed 4 years ago

nile649 commented 4 years ago

Error(s) in loading state_dict for AOGNet: Missing key(s) in state_dict: "stem.0.

nile649 commented 4 years ago

apparently i replaced 15 the index, from 8 in the loader to make it work.