Train the model

[Auth0rM0rgan]

Hey @yu45020,

I was trying to train the model on my own dataset but I couldn't. It would be nice if you can provide the instructions to use your code for training and testing on another dataset.

In addition, please check the train.py since there is not apex file to import it. from apex.fp16_utils import FP16_Optimizer, network_to_half

Thanks in advance!

[yu45020]

You may get apex from here, which can be used in both windows and linux. But if you do inference in CPU only, you must use fp32 because PyTorch doesn't support fp16 at this time.

Let me try to complete the ReadME and add checkpoints on the most recent model this week .

Thanks for your interest.

[yu45020]

@Auth0rM0rgan

I add the most recent model checkpoint and ways to resume it for further training. Hope it helps.