configs/small.json

{"width":64,"depth":4,"modes":12,"rank":16,"batch_size":128,"epochs":30,"lr":0.0003,"weight_decay":0.0001,"workers":4}