Training it on custom dataset
humzaahmad906 opened this issue · comments
Humza Ahmad commented
I want to train the given network on my own dataset. What steps do i need to follow?
Petrov Ilya commented
You have to define dataset class for your data (see SBDDataset for an example) and then create a model training script for your dataset (see r34_dh128 for an example).
Humza Ahmad commented
while training the network on custom dataset it gets stuck. Can you help me with it?