{ "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 }