Skip to content

Commit e83e6b3

Browse files
authored
Update kd.py
1 parent 82a9a97 commit e83e6b3

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

scripts/kd.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -210,16 +210,16 @@ def main():
210210

211211

212212
# path to the 1st teacher model
213-
PATH_teacher_models = '/home/dl33629@austin.utexas.edu/research-lowhigh/models/cnn14+fc/sr_%d_shuffle/seg_%ds/models_for_esc_50class/%s/' %(sr_teacher, seg_t, sub) # for esc: models_for_esc_50class
214-
#PATH_teacher_models = '/home/dl33629@austin.utexas.edu/research-lowhigh/models/mbnet+fc/student/esc50_50class/FineTuneMBNetV2/sr_%d_1teacher/seg_5s/%s/' %(1000, sub) # for multi stage
213+
PATH_teacher_models = '/home/research-lowhigh/models/cnn14+fc/sr_%d_shuffle/seg_%ds/models_for_esc_50class/%s/' %(sr_teacher, seg_t, sub) # for esc: models_for_esc_50class
214+
#PATH_teacher_models = '/home/research-lowhigh/models/mbnet+fc/student/esc50_50class/FineTuneMBNetV2/sr_%d_1teacher/seg_5s/%s/' %(1000, sub) # for multi stage
215215
# path to the 2nd teacher model, if any
216-
#PATH_teacher_models2 = '/home/dl33629@austin.utexas.edu/research-lowhigh/models/mbnet+fc/sr_%d/seg_%ds/models_for_adl/%s/' %(16000, seg_t, sub)
216+
#PATH_teacher_models2 = '/home/research-lowhigh/models/mbnet+fc/sr_%d/seg_%ds/models_for_adl/%s/' %(16000, seg_t, sub)
217217
PATH_teacher_models2 = None
218218
teacher_model_name = 'FineTuneCNN14' #'FineTuneCNN14', 'fc_3layers', 'FineTuneMBNetV2', 'FineTuneResNet38'
219219
student_model_name = 'FineTuneCNN14' #'FineTuneCNN14', 'fc_3layers', 'FineTuneMBNetV2', 'FineTuneResNet38'
220220
# adl and esc share the same path to save checkpoints / logs
221-
PATH_save_models = '/home/dl33629@austin.utexas.edu/research-lowhigh/models/cnn14+fc/student/esc50_50class/%s/sr_%d/seg_%ds/%s/' %(student_model_name, sr_student, seg_t, sub)
222-
PATH_log = '/home/dl33629@austin.utexas.edu/research-lowhigh/models/cnn14+fc/student/esc50_50class/%s/sr_%d/seg_%ds/%s/' %(student_model_name, sr_student, seg_t, sub)
221+
PATH_save_models = '/home/research-lowhigh/models/cnn14+fc/student/esc50_50class/%s/sr_%d/seg_%ds/%s/' %(student_model_name, sr_student, seg_t, sub)
222+
PATH_log = '/home/research-lowhigh/models/cnn14+fc/student/esc50_50class/%s/sr_%d/seg_%ds/%s/' %(student_model_name, sr_student, seg_t, sub)
223223

224224
# prepare teacher model
225225
teacher_model = load_teacher_model(PATH_teacher_models, teacher_model_name, classes_num)

0 commit comments

Comments
 (0)