Not getting good result after training #90
Description
Actually I have prepared my own data set of indoor scene in my environment and want to train model on that. I am freezing all other layers except for the up projection blocks and the result is not so good. Even I trained it on as small data set as 600 images and achieved 82 percent accuracy but the results were not good visually.
I donot know the reason of that maybe you can suggest me something.
And the images I want to train are approximately 6k.
The pretrained weights with NYU are even performing better.
batch_size = 32
learning_rate = 1.0e-3
monentum = 0.9
weight_decay = 0.0005
num_epochs = 70
optimizer = torch.optim.SGD(filter(lambda p: p.requires_grad, model.parameters()), lr=learning_rate,
momentum=monentum, weight_decay=weight_decay)
and lr is halved after 10 epochs.
Validation depth image
rgb image
