diff --git a/model/src/loss_fn.py b/model/src/loss_fn.py deleted file mode 100644 index fe84786..0000000 --- a/model/src/loss_fn.py +++ /dev/null @@ -1,7 +0,0 @@ -from __future__ import annotations - -import torch.nn as nn - - -def setup_criterion(): - return nn.MSELoss()