The problem of training.

ChenhongyiYang / PlainMamba

[BMVC 2024] PlainMamba: Improving Non-hierarchical Mamba in Visual Recognition

Apache License 2.0

68 stars 7 forks source link

The problem of training. #2

Closed LIUYUANWEI98 closed 6 months ago

LIUYUANWEI98 commented 6 months ago

When training the model, I met the error ’AmpEpochBasedRunner should be used with grad_clip‘. How can I fix it?

LIUYUANWEI98 commented 6 months ago

When I am ready to train PlainMamba-L1 model, I submitted 'sh tool/dist_train.sh plain_mamba_configs/plain_mamba_l1_in1k_300e.py 4' and met the error like the follow:

ChenhongyiYang commented 6 months ago

Hi, Can you plz try changing AmpEpochBasedRunner to EpochBasedRunner?