diff --git a/configs/config_EF_AC_Szada.yaml b/configs/config_EF_AC_Szada.yaml index 0c9b9172fa4d2efea38ed74b124b6f1f2f8773fa..8579c5976b84fc60dac8a565f73ac71545f3fdbf 100644 --- a/configs/config_EF_AC_Szada.yaml +++ b/configs/config_EF_AC_Szada.yaml @@ -24,7 +24,7 @@ resume: '' load_optim: True save_optim: True anew: False -trace_freq: 1 +track_intvl: 1 device: cuda metrics: 'F1Score+Accuracy+Recall+Precision' diff --git a/configs/config_EF_AC_Tiszadob.yaml b/configs/config_EF_AC_Tiszadob.yaml index 91632ddce86b521d5f860551e13032a2058ca24e..52a103abff56a490c1c57ea2c0bc5f9f9de9d4a6 100644 --- a/configs/config_EF_AC_Tiszadob.yaml +++ b/configs/config_EF_AC_Tiszadob.yaml @@ -24,7 +24,7 @@ resume: '' load_optim: True save_optim: True anew: False -trace_freq: 1 +track_intvl: 1 device: cuda metrics: 'F1Score+Accuracy+Recall+Precision' diff --git a/configs/config_EF_OSCD.yaml b/configs/config_EF_OSCD.yaml index 9902142254332ac316c750523f4b0c489cf40833..bb1005eeea36bdbf0a7f2205db994571bd8a820b 100644 --- a/configs/config_EF_OSCD.yaml +++ b/configs/config_EF_OSCD.yaml @@ -24,7 +24,7 @@ resume: '' load_optim: True save_optim: True anew: False -trace_freq: 1 +track_intvl: 1 device: cuda metrics: 'F1Score+Accuracy+Recall+Precision' diff --git a/configs/config_base.yaml b/configs/config_base.yaml index f1b5ae959b85cab26407821c3706ed40efd39c85..2e1119b044f3feea04a24b70001003fd59a15a0a 100644 --- a/configs/config_base.yaml +++ b/configs/config_base.yaml @@ -24,7 +24,7 @@ resume: '' load_optim: True save_optim: True anew: False -trace_freq: 1 +track_intvl: 1 device: cuda metrics: 'F1Score+Accuracy+Recall+Precision' diff --git a/configs/config_siamconc_AC_Szada.yaml b/configs/config_siamconc_AC_Szada.yaml index 5be549448d7570dd2a082d7e5af720d28a5f9cea..d1972fba51cbddb431840fedb199002e3224ccec 100644 --- a/configs/config_siamconc_AC_Szada.yaml +++ b/configs/config_siamconc_AC_Szada.yaml @@ -24,7 +24,7 @@ resume: '' load_optim: True save_optim: True anew: False -trace_freq: 1 +track_intvl: 1 device: cuda metrics: 'F1Score+Accuracy+Recall+Precision' diff --git a/configs/config_siamconc_AC_Tiszadob.yaml b/configs/config_siamconc_AC_Tiszadob.yaml index 979facfe47a19d064f2cd0707898ea781f727af5..1ab90f9d087e6b59a700aae4ad131476f5b7a50b 100644 --- a/configs/config_siamconc_AC_Tiszadob.yaml +++ b/configs/config_siamconc_AC_Tiszadob.yaml @@ -24,7 +24,7 @@ resume: '' load_optim: True save_optim: True anew: False -trace_freq: 1 +track_intvl: 1 device: cuda metrics: 'F1Score+Accuracy+Recall+Precision' diff --git a/configs/config_siamconc_OSCD.yaml b/configs/config_siamconc_OSCD.yaml index 640f83dfe48257b7dbaa3dad7efd0b6cd586ac81..a007e1ea38a343993f1b85f55c68b39dd0adb93b 100644 --- a/configs/config_siamconc_OSCD.yaml +++ b/configs/config_siamconc_OSCD.yaml @@ -24,7 +24,7 @@ resume: '' load_optim: True save_optim: True anew: False -trace_freq: 1 +track_intvl: 1 device: cuda metrics: 'F1Score+Accuracy+Recall+Precision' diff --git a/configs/config_siamdiff_AC_Szada.yaml b/configs/config_siamdiff_AC_Szada.yaml index 055f97092c06b4ed72973f65f4861a9fca72e2ff..e569a2bdafaad1e39134c29474412846b134800f 100644 --- a/configs/config_siamdiff_AC_Szada.yaml +++ b/configs/config_siamdiff_AC_Szada.yaml @@ -24,7 +24,7 @@ resume: '' load_optim: True save_optim: True anew: False -trace_freq: 1 +track_intvl: 1 device: cuda metrics: 'F1Score+Accuracy+Recall+Precision' diff --git a/configs/config_siamdiff_AC_Tiszadob.yaml b/configs/config_siamdiff_AC_Tiszadob.yaml index 931fc0c49fe07eb7afb349807e1f763d3e07acab..5d6a35d0c5491f30dc209f45709f1ecb8ea75e38 100644 --- a/configs/config_siamdiff_AC_Tiszadob.yaml +++ b/configs/config_siamdiff_AC_Tiszadob.yaml @@ -24,7 +24,7 @@ resume: '' load_optim: True save_optim: True anew: False -trace_freq: 1 +track_intvl: 1 device: cuda metrics: 'F1Score+Accuracy+Recall+Precision' diff --git a/configs/config_siamdiff_OSCD.yaml b/configs/config_siamdiff_OSCD.yaml index ca28ae30da2981461b07152eb2c132256233c332..c871805b20b4a790e8d354a885e312e87fa7c650 100644 --- a/configs/config_siamdiff_OSCD.yaml +++ b/configs/config_siamdiff_OSCD.yaml @@ -24,7 +24,7 @@ resume: '' load_optim: True save_optim: True anew: False -trace_freq: 1 +track_intvl: 1 device: cuda metrics: 'F1Score+Accuracy+Recall+Precision' diff --git a/src/core/trainers.py b/src/core/trainers.py index 83330b3be8ef599b7d9390a8a5852f8651e50d16..68cde97c53b6608b0a95d8723b89487dc56e6658 100644 --- a/src/core/trainers.py +++ b/src/core/trainers.py @@ -33,7 +33,7 @@ class Trainer: self.lr = float(context.lr) self.save = context.save_on or context.out_dir self.out_dir = context.out_dir - self.trace_freq = int(context.trace_freq) + self.track_intvl = int(context.track_intvl) self.device = torch.device(context.device) self.suffix_off = context.suffix_off @@ -188,7 +188,7 @@ class Trainer: } # Save history history_path = self.path('weight', constants.CKP_COUNTED.format(e=epoch), underline=True) - if epoch % self.trace_freq == 0: + if epoch % self.track_intvl == 0: torch.save(state, history_path) # Save latest latest_path = self.path( diff --git a/src/train.py b/src/train.py index 782336b048b55f8e0345f84f47d566c92fe6a90d..c27895408daa6dc1cd94639a1e64ce022b9bbfdd 100644 --- a/src/train.py +++ b/src/train.py @@ -67,7 +67,7 @@ def parse_args(): help='path to latest checkpoint') group_train.add_argument('--anew', action='store_true', help='clear history and start from epoch 0 with the checkpoint loaded') - group_train.add_argument('--trace-freq', type=int, default=50) + group_train.add_argument('--track_intvl', type=int, default=50) group_train.add_argument('--device', type=str, default='cpu') group_train.add_argument('--metrics', type=str, default='F1Score+Accuracy+Recall+Precision')