ynet icon indicating copy to clipboard operation
ynet copied to clipboard

Training issue

Open Akhillalpp opened this issue 1 year ago • 1 comments

During training, if testset evaluation is removed, then model returns a different result. That is if we Remove lines

print("Expert 1 - Test") dice_test = eval(test_loader, criterion_seg, model, n_classes=n_classes)

from the below code, training result is different.

        if t % 10 == 0 or t > 45:
        print("Epoch", t, "/", iterations)
        print("Validation")
        dice = eval(val_loader, criterion_seg, model, dice_s=True, n_classes=n_classes)
        print("Expert 1 - Test")
        dice_test = eval(test_loader, criterion_seg, model, n_classes=n_classes)

Is it implies the evaluation affects the model parameters? That's why we are getting a different result?
model.eval() is given in eval function. But still results are different. Why?

Akhillalpp avatar Oct 21 '24 06:10 Akhillalpp

Found same issue. Can anyone clarify on this

Deepthi-vr-code avatar Oct 21 '24 07:10 Deepthi-vr-code