diff --git a/MODEL.pth.REMOVED.git-id b/MODEL.pth.REMOVED.git-id new file mode 100644 index 0000000..8f9fb71 --- /dev/null +++ b/MODEL.pth.REMOVED.git-id @@ -0,0 +1 @@ +408f675eb803bd50727626d588144df3f99e6234 \ No newline at end of file diff --git a/train.py b/train.py index c09d1d3..59dd2c1 100644 --- a/train.py +++ b/train.py @@ -22,7 +22,6 @@ def train_net(net, epochs=5, batch_size=2, lr=0.1, val_percent=0.05, dir_mask = 'data/train_masks/' dir_checkpoint = 'checkpoints/' - # get ids ids = get_ids(dir_img) ids = split_ids(ids)