Initial commit
Browse files- README.md +10 -0
- config.json +1 -0
- pytorch_model.bin +3 -0
- spm.model +3 -0
README.md
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language: en
|
| 3 |
+
tags:
|
| 4 |
+
- summarization
|
| 5 |
+
inference: false
|
| 6 |
+
license: bsd-3-clause
|
| 7 |
+
---
|
| 8 |
+
|
| 9 |
+
## BiMeanVAE model
|
| 10 |
+
See original GitHub repo for more details [here](https://github.com/megagonlabs/coop)
|
config.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"data_dir": "./data/yelp", "model": {"embedding_dim": 256, "free_bit": 0.25, "hidden_size": 512, "latent_dim": 512, "num_layers": 1, "type": "bimeanvae"}, "spm_path": "./data/sentencepiece/yelp.model", "trainer": {"batch_size": 256, "checkout_step": 1000, "clip_value": 5, "lr": 0.001, "num_steps": 100000}}
|
pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8b700125095aaa4879f9f0e26370966ed9dabb6a6241f8478c4215427ab033fc
|
| 3 |
+
size 52342308
|
spm.model
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5f8c33921294e1b2cb8b1c9a133782b2ed4239633c21b40bde8e2c066a82d910
|
| 3 |
+
size 775519
|