args.json

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