How to save the trained model

Question

How to save the trained model

peyer opened this issue 8 years ago · comments

peyer commented 8 years ago

If I want to save the pnet/rnet/onet model , should I add mod.save(...) after mod.fit(...) in train.py?

Xuan Lin · Answer 1 · Wed Feb 15 2017 22:11:28 GMT+0800 (China Standard Time)

No, it's done in epoch end callback inside mod.fit()

peyer · Answer 2 · Wed Feb 15 2017 22:20:37 GMT+0800 (China Standard Time)

thanks a lot

peyer · Answer 3 · Thu Feb 16 2017 09:22:40 GMT+0800 (China Standard Time)

I'm sorry , but when I train the pnet again using your gen_pnet_data.py and lr = 0.01 , the output become nan again . I feel so bad .

zhangfeng13 · Answer 4 · Sat May 27 2017 23:46:16 GMT+0800 (China Standard Time)

@peyer can you share me the blog about how to train the Mtcnn.

anna0709 · Answer 5 · Fri Dec 01 2017 15:59:22 GMT+0800 (China Standard Time)

@peyer Hi, I am new in mxnet, and I want to retrain P and R and net on the dataset of wider face. I had run transform.py to get anno.txt, then run gen_pnet_data.py and gen_hard_example.py, what should I do next? run train_P_net.py ? could you please tell me the whole step of training?