bimeanvae-amzn / config.json
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}}
2