wp99cp/bachelor-thesis

This includes a Softmax within the loss function

Closed this issue a year ago · 0 comments

wp99cp commented a year ago

Thus we should remove our softmax form the model?

bachelor-thesis/models/unet/model/training.py

Line 42 in 09fbe5f

lossFunc = BCEWithLogitsLoss(pos_weight=class_weights.to(DEVICE))