| { | |
| "dataset_name": "Movies_and_TV", | |
| "sequence_length": 100, | |
| "embedding_dim": 64, | |
| "num_heads": 1, | |
| "max_batches_per_epoch": 500, | |
| "num_blocks": 2, | |
| "dropout_rate": 0.5, | |
| "negs_per_pos": 256, | |
| "gbce_t": 0.75, | |
| "state_dict_path": "ML/models/saved_models/gsasrec-Movies_and_TV-step=326197.pt", | |
| "usernum": 311143, | |
| "itemnum": 86678, | |
| "device": "cuda:0" | |
| } |