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/amzn", "model": {"embedding_dim": 256, "free_bit": 0.25, "hidden_size": 512, "latent_dim": 512, "num_layers": 1, "type": "bimeanvae"}, "spm_path": "./data/sentencepiece/amzn.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:b5eeddd56c397c19bba9afd93bcee66f8dd8f55b699e959784bf491e74886d3a
|
| 3 |
+
size 52342308
|
spm.model
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7a41fd15bc4ea778dfccd18b22eb1aa4b3aabaa02fa2febdf3757da68a894ac3
|
| 3 |
+
size 770974
|