Yajneshwar Mandal

Hello sir, Thank you for your great work. I want to re-train your existing model-32 with my custom data set . For that i am replacing self.load_state_dict(torch.load('../res/resnet50-19c8e357.pth'), strict=False) with self.load_state_dict(torch.load('../res/model-32'),...

Error after few iteration on training

5

I want to train the network --arch 7 with my custom 62k dataset that is similar to DUTS. I am using 48GB CUDA and batch size 8. After a few...

Train with my custom large dataset

3

I have a large dataset of 300K images with different objects,car,human, animals, etc. Should I need to train from scratch or I can do transfer learning from your general pre-trained...