search

GeWu-Lab / OGM-GE_CVPR2022

The repo for "Balanced Multimodal Learning via On-the-fly Gradient Modulation", CVPR 2022 (ORAL)
MIT License
236 stars 18 forks source link

main.py #11

Closed Wbojiao closed 2 years ago

Wbojiao commented 2 years ago

338 model = model.load_state_dict(state_dict) 语法似乎错误，该改为model.load_state_dict(state_dict)