Implementation of Deep Learning unit tests, as well as paths to the 2d data for windows users in the UNet jupyter notebook.
Compare changes
- OskarK authored
+ 6
− 3
@@ -66,13 +66,15 @@ def train_model(model, hyperparameters, train_loader, val_loader, eval_every = 1
@@ -99,7 +101,7 @@ def train_model(model, hyperparameters, train_loader, val_loader, eval_every = 1
@@ -206,7 +208,8 @@ def inference(data,model):