-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsample_launch.json
105 lines (105 loc) · 3.25 KB
/
sample_launch.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
{
// Use IntelliSense to learn about possible attributes.
// Hover to view descriptions of existing attributes.
// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
"version": "0.2.0",
"configurations": [
{
"name": "Python: mymodel",
"type": "python",
"request": "launch",
"program": "mymodel.py",
"console": "integratedTerminal",
"justMyCode": false,
"cwd": "${workspaceFolder}/ensemble_model/ensemble_model_emb"
},
{
"args": [
"--device",
"cuda:7",
"--num_epochs",
"10000",
"--dataset",
"ml-1m",
"--save_dir",
"saved",
"--state_dict_path",
"saved/SASRec.epoch=10000.lr=0.001.layer=2.head=1.hidden=64.maxlen=200.reg_weight=0.0.pth",
"--inference_only",
"true",
"--maxlen",
"200",
"--hidden_units",
"64"
],
"name": "Python: train_SASRec",
"type": "python",
"request": "launch",
"program": "train_SASRec.py",
"console": "integratedTerminal",
"justMyCode": false,
"cwd": "${workspaceFolder}/base_model"
},
{
"args": [
"--device",
"cuda:3",
"--epochs",
"1",
"--datapath",
"dataset/movielens1m_m1",
"--save_dir",
"saved",
"--decay",
"1e-4",
"--lr",
"0.001",
"--layer",
"3",
"--seed",
"2020",
"--dataset",
"ml-1m",
"--topks",
"[10]",
"--recdim",
"64",
"--pop_reg_weight",
"0.0"
],
"name": "Python: train_LightGCN",
"type": "python",
"request": "launch",
"program": "train_LightGCN.py",
"console": "integratedTerminal",
"justMyCode": false,
"cwd": "${workspaceFolder}/base_model"
},
{
"args": [
"--config",
"./config/SimpleX_movielens1m_m1",
"--expid",
"SimpleX_movielens1m_m1",
"--gpu",
"6"
],
"name": "Python: train_SimpleX",
"type": "python",
"request": "launch",
"program": "train_SimpleX.py",
"console": "integratedTerminal",
"justMyCode": false,
"cwd": "${workspaceFolder}/base_model"
},
// {
// "args": ["--config ./config/SimpleX_movielens1m_m1", "--expid SimpleX_movielens1m_m1 --gpu 6"],
// "name": "Python: train_SimpleX",
// "type": "python",
// "request": "launch",
// "program": "train_SimpleX.py",
// "console": "integratedTerminal",
// "justMyCode": false
// }
]
}