Unable to load the trained checkpoints

[fudanvi]

Unable to load the trained checkpoints

[kanika02]

In fact, it works in my environment ... so have you modified the code or made some big changes?

No I have not made any change

[JingyeChen]

Please see Readme.md
CUDA_VISIBLE_DEVICES=GPU_NUM python main.py --batch_size=16 --STN --exp_name EXP_NAME --text_focus 【--resume YOUR_MODEL】 --test --test_data_dir ./dataset/mydata/test

or you can fill the absolute address of the pre-trained model in the argpaser of main.py

Hope this advice will help you out.

[kanika02]

My epoch got interrupted and i want to resume it from the last interrupted so that I don't have to start my epoch from the starting (that will take a lot of my time) . Please help me to resume that . I tried but always cleans the old logs and start from the starting epochs

[JingyeChen]

Hello, if you want to resume the checkpoint, you could [modify the name] of the experiment to solve this problem.

[kanika02]

!python main.py --batch_size=16 --STN --exp_name f2 --text_focus --resume /content/drive/MyDrive/scene-text-telescope/checkpoint/f1/checkpoint.pth

i have used this command (changed the name ) still the training is starting from the first epochs

[JingyeChen]

It is quite weird... So have you tried to directly modify the resume path in main.py?

[kanika02]

Yes, I have modified the main.py, still not working. The epochs always start from the starting I have tried all ways. Please help me because I am not able to train the model for large epochs.

[JingyeChen]

In fact, it works in my environment ...
so have you modified the code or made some big changes?

[kanika02]

my epochs got interrupted here then I tried to resume it from this point but it starts from the starting like this .

Tried everything you told still not working.
can you share your environment so that I can try in that environment

[Uxasxie]

i got the same problem.do you solve it? im so disgusting

[cptbtptp125]

Hello, could you tell me how to solve this problem? I have also met the same situation. Resume doesn't work at all