From 932e1c3cf53e2661e4f94a89b7e73500d398e7ed Mon Sep 17 00:00:00 2001 From: milesial Date: Wed, 23 Aug 2017 17:38:55 +0200 Subject: [PATCH] Added simple trained model Former-commit-id: e87526c043c447e5c8d95ee8314fcb7f97c13ce2 --- MODEL.pth.REMOVED.git-id | 1 + train.py | 1 - 2 files changed, 1 insertion(+), 1 deletion(-) create mode 100644 MODEL.pth.REMOVED.git-id 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)