how to train the net on my own dataset

tonylins / pytorch-mobilenet-v2

A PyTorch implementation of MobileNet V2 architecture and pretrained model.

Apache License 2.0

1.37k stars 328 forks source link

how to train the net on my own dataset #21

Closed JimReno closed 6 years ago

JimReno commented 6 years ago

Pretty good job tony！Thanks for the brilliant work you've done! I am a novice in DL and just wondering how to train the network on my own dataset. Where is the “”input data“” snippet：）

tonylins commented 6 years ago

I did not provide imagenet data loader here. Please refer to the Pytorch official tutorial. Thanks!