search

DonaldRR / SimpleNet

MIT License
402 stars 59 forks source link

how to predict #31

Closed houruixiang-1024 closed 8 months ago

houruixiang-1024 commented 11 months ago

I haven't found any weight files saved after training. How to use the already trained weights for prediction

DonaldRR commented 8 months ago

I'm sure it is saved in the "train" method.

DonaldRR commented 8 months ago

It is saved with the "torch.load". More info about saving/loading pls refer to Torch's doc