ENSEEIHT/projet-long
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Added simple trained model

Browse source 
Former-commit-id: e87526c043c447e5c8d95ee8314fcb7f97c13ce2
This commit is contained in:
milesial 2017-08-23 17:38:55 +02:00
parent 74ee006a7a
commit 932e1c3cf5
2 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
MODEL.pth.REMOVED.git-id Normal file
View file

@ -0,0 +1 @@
408f675eb803bd50727626d588144df3f99e6234

1
train.py
View file

@ -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)