metricv
/

metricsubs-segmenter

Model card Files Files and versions Community

metricv commited on Aug 31

Commit

5414abe

•

1 Parent(s): 64eaa7b

Update model

Files changed (3) hide show

data +1 -1
segmenter.ckpt +1 -1
train.py +1 -1

data CHANGED Viewed

	@@ -1 +1 @@
1	- Subproject commit ~~83ccdae5afe7eaf7f88b0ceb4933544e445b7841~~


1	+ Subproject commit 33c57a3cafbdb46b4cc7db7f08695d63b52d6668

segmenter.ckpt CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e616108c71de535ed24386f1aaf2e38ab9cdf0dd123517aa5cef72c9ec019ed9
 size 2665888

 version https://git-lfs.github.com/spec/v1
+oid sha256:26c09246a1ed23aa5be9656e36878d30b1b39aa649dbd9a24bbef7ecee5a4e7d
 size 2665888

train.py CHANGED Viewed

@@ -26,6 +26,6 @@ if __name__ == "__main__":
     model.to(device)
-    train_bidirlstm_embedding_model(model, dataset, num_epochs=150, batch_size=2)
     torch.save(model.state_dict(), "segmenter.ckpt")

     model.to(device)
+    train_bidirlstm_embedding_model(model, dataset, num_epochs=75, batch_size=2)
     torch.save(model.state_dict(), "segmenter.ckpt")