-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathtest.py
More file actions
44 lines (42 loc) · 3.3 KB
/
test.py
File metadata and controls
44 lines (42 loc) · 3.3 KB
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
import os
import json
import numpy as np
run_inferece = True
continual_dict = {
"0camel": dict(
config="/data6/doanhbc/continual_learning_project/7.continual-learning_fold2_2/configs_cesc_c16/cesc.yaml",
checkpoint="/data6/doanhbc/continual_learning_project/7.continual-learning_no_task_indicator_reverse/logs_dir_continual_learning/logs_cesc/configs/cesc/fold0/epoch=04-val_loss=0.0237.ckpt"
),
"0camel_1nsclc": dict(
config="/data6/doanhbc/continual_learning_project/7.continual-learning_fold2_2/configs_cesc_c16/cesc_tgct.yaml",
checkpoint="/data6/doanhbc/continual_learning_project/7.continual-learning_no_task_indicator_reverse/logs_dir_continual_learning/logs_cesc_tgct/configs/cesc_tgct/fold0/epoch=11-val_loss=0.0159.ckpt"
),
"0camel_1nsclc_2brca": dict(
config="/data6/doanhbc/continual_learning_project/7.continual-learning_fold2_2/configs_cesc_c16/cesc_tgct_esca.yaml",
checkpoint="/data6/doanhbc/continual_learning_project/7.continual-learning_no_task_indicator_reverse/logs_dir_continual_learning/logs_cesc_tgct_esca/configs/cesc_tgct_esca/fold0/epoch=44-val_loss=0.0011.ckpt"
),
"0camel_1nsclc_2brca_3rcc": dict(
config="/data6/doanhbc/continual_learning_project/7.continual-learning_fold2_2/configs_cesc_c16/cesc_tgct_esca_rcc.yaml",
checkpoint="/data6/doanhbc/continual_learning_project/7.continual-learning_no_task_indicator_reverse/logs_dir_continual_learning/logs_cesc_tgct_esca_rcc/configs/cesc_tgct_esca_rcc/fold0/epoch=00-val_loss=0.0890.ckpt"
),
"0camel_1nsclc_2brca_3rcc_4esca": dict(
config="/data6/doanhbc/continual_learning_project/7.continual-learning_fold2_2/configs_cesc_c16/cesc_tgct_esca_rcc_brca.yaml.yaml",
checkpoint="/data6/doanhbc/continual_learning_project/7.continual-learning_no_task_indicator_reverse/logs_dir_continual_learning/logs_cesc_tgct_esca_rcc_brca/configs/cesc_tgct_esca_rcc_brca/fold0/epoch=01-val_loss=0.1104.ckpt"
),
"1.CAMEL_2.NSCLC_3.BRCA_4.RCC_5.ESCA_6.TGCT": dict(
config="/data6/doanhbc/continual_learning_project/7.continual-learning_fold2_2/configs_cesc_c16/cesc_tgct_esca_rcc_brca_nsclc.yaml",
checkpoint="/data6/doanhbc/continual_learning_project/7.continual-learning_no_task_indicator_reverse/logs_dir_continual_learning/logs_cesc_tgct_esca_rcc_brca_nsclc/configs/cesc_tgct_esca_rcc_brca_nsclc/fold0/epoch=03-val_loss=0.0720.ckpt"
),
"1.CAMEL_2.NSCLC_3.BRCA_4.RCC_5.ESCA_6.TGCT_7.CESC": dict(
config="/data6/doanhbc/continual_learning_project/7.continual-learning_fold2_2/configs_cesc_c16/cesc_tgct_esca_rcc_brca_nsclc_camel.yaml",
checkpoint="/data6/doanhbc/continual_learning_project/7.continual-learning_no_task_indicator_reverse/logs_dir_continual_learning/logs_cesc_tgct_esca_rcc_brca_nsclc_camel/configs/cesc_tgct_esca_rcc_brca_nsclc_camel/fold0/epoch=09-val_loss=0.0045.ckpt"
)
}
if run_inferece:
for sequence_data in continual_dict:
config_path = continual_dict[sequence_data]['config']
checkpoint_path = continual_dict[sequence_data]['checkpoint']
command_inference = \
"CUDA_VISIBLE_DEVICES=4 python -W ignore train.py --stage=test --config={config} --model_path={checkpoint} --gpu=0".format(
config=config_path, checkpoint=checkpoint_path)
os.system(command_inference)