Training on custom dataset

[loveis98]

Hello! Will 12 GB of single GPU be enough to train your model on a small dataset? And may be you can publish your pretrained model (dataset no matter)? It would be great)

[wuhuikai]

1. The pre-trained model can be found here.
2. A single GPU is enough if you freeze all the BN layers.

[meanmee]

@wuhuikai could you show us how to make a custom dataset to fit into the training process? what is the data structure?

[wuhuikai]

Please refer to the Base class and a concrete class.