config.json

{
    "dataset": "Movies_and_TV",
    "train_dir": "default",
    "batch_size": 200,
    "lr": 0.001,
    "maxlen": 50,
    "hidden_units": 64,
    "num_blocks": 2,
    "num_epochs": 1000,
    "num_heads": 1,
    "dropout_rate": 0.2,
    "l2_emb": 5e-05,
    "device": "cuda",
    "inference_only": true,
    "state_dict_path": "Movies_and_TV_default/TiSASRec_model_epoch_1000.pt",
    "time_span": 256,
    "usernum": 311143,
    "itemnum": 86678
}