megagonlabs
/

optimus-amzn

Model card Files Files and versions Community

optimus-amzn / config.json

isomap's picture

Initial commit

af204b4 over 3 years ago

197 Bytes

	{"data_dir": "./data/amzn", "model": {"free_bit": 2, "latent_dim": 512, "type": "optimus"}, "trainer": {"batch_size": 4, "checkout_step": 20000, "clip_value": 5, "lr": 1e-05, "num_steps": 500000}}