Hugging Face's logo

PNUDI
/
gSASRec

PyTorch
Model card Files
xet
Community
knh123 commited on
Commit
616186d
·
verified ·
1 Parent(s): 3e51a41

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +14 -12
config.json CHANGED
@@ -1,13 +1,15 @@
1
- {
2
- "dataset_name": "Movies_and_TV",
3
- "sequence_length": 100,
4
- "embedding_dim": 64,
5
- "num_heads": 1,
6
- "max_batches_per_epoch": 500,
7
- "num_blocks": 2,
8
- "dropout_rate": 0.5,
9
- "negs_per_pos": 256,
10
- "gbce_t": 0.75,
11
- "state_dict_path": "ML/models/saved_models/gsasrec-Movies_and_TV-step=326197.pt",
12
- "device": "cuda:0"
 
 
13
  }
 
1
+ {
2
+ "dataset_name": "Movies_and_TV",
3
+ "sequence_length": 100,
4
+ "embedding_dim": 64,
5
+ "num_heads": 1,
6
+ "max_batches_per_epoch": 500,
7
+ "num_blocks": 2,
8
+ "dropout_rate": 0.5,
9
+ "negs_per_pos": 256,
10
+ "gbce_t": 0.75,
11
+ "state_dict_path": "ML/models/saved_models/gsasrec-Movies_and_TV-step=326197.pt",
12
+ "usernum": 311143,
13
+ "itemnum": 86678,
14
+ "device": "cuda:0"
15
  }