loss.py only understands monai losses, but not MSELoss

kbressem / trainlib

MIT License

4 stars 0 forks source link

loss.py only understands monai losses, but not MSELoss #37

Closed markus-hinsche closed 1 year ago

markus-hinsche commented 2 years ago

https://github.com/kbressem/trainlib/blob/main/trainlib/loss.py

MSELoss is defined in torch.nn

kbressem commented 2 years ago

I'll rewrite the loss.py to accept more general loss functions. But until then, try patching loss.py. Loss, model, optim and transforms can all be overwritten by patch.