| { | |
| "dataset": "ml-1m", | |
| "train_dir": "colab_run", | |
| "batch_size": 128, | |
| "lr": 0.001, | |
| "maxlen": 200, | |
| "hidden_units": 50, | |
| "num_blocks": 2, | |
| "num_epochs": 1000, | |
| "num_heads": 1, | |
| "dropout_rate": 0.2, | |
| "l2_emb": 0.0, | |
| "device": "cuda", | |
| "inference_only": false, | |
| "state_dict_path": null, | |
| "norm_first": false | |
| } |