model

[TIAN-Ww]

Hi, In the code you provided, I did not find the code to save the trained model. Can you help me?
Best wishes

[TIAN-Ww]

I encountered this error during training. Do I need to have a model before training? Looking forward to your reply!
Best wishes！

[JohnleeHIT]

I encountered this error during training. Do I need to have a model before training? Looking forward to your reply! Best wishes！

Don't worry, actually, this is not a error. This is used when you have to train your model based on an older checkpoint, if you don't have any, you can still train from scratch, It is totally fine.

[JohnleeHIT]

Hi, In the code you provided, I did not find the code to save the trained model. Can you help me? Best wishes

Brats2019/src/operations.py

Line 744 in 77418bc

def save_chkpoint(self, checkpoint_dir, model_name, step):