forked from JuneQiong/PESI
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathTuneSeed.py
More file actions
20 lines (13 loc) · 727 Bytes
/
Copy pathTuneSeed.py
File metadata and controls
20 lines (13 loc) · 727 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import os
import pdb
seed = [0, 11, 111, 1111]
print('test lr con lr on amo dataset')
count = 0
for se in seed:
checkpoint = 'tuneparam_amoclo_seed_{}'.format(se)
outf ='tuneparam_amoclo_seed_{}'.format(se)
print('-----begin---main_init.py --seed{}'.format(se))
# os.system(
# 'python main_init.py --checkpoint={} --outf={} --lr={} --con_lr={} --rating_reg={} --con_reg={} --conself_reg={}'.format(checkpoint, outf, l, cl, rr, cr, cr))
os.system('python main_init.py --seed %s --checkpoint %s --outf %s'%(se, checkpoint, outf))
print('-----end---main_init.py --seed_{}'.format(se))