From e80eae38d18e8eab3d1e87f0b04d143de692aeda Mon Sep 17 00:00:00 2001 From: Bobholamovic <bob1998425@hotmail.com> Date: Thu, 10 Sep 2020 10:33:34 +0800 Subject: [PATCH] Change unclear statement: trace_freq to track_intvl --- configs/config_EF_AC_Szada.yaml | 2 +- configs/config_EF_AC_Tiszadob.yaml | 2 +- configs/config_EF_OSCD.yaml | 2 +- configs/config_base.yaml | 2 +- configs/config_siamconc_AC_Szada.yaml | 2 +- configs/config_siamconc_AC_Tiszadob.yaml | 2 +- configs/config_siamconc_OSCD.yaml | 2 +- configs/config_siamdiff_AC_Szada.yaml | 2 +- configs/config_siamdiff_AC_Tiszadob.yaml | 2 +- configs/config_siamdiff_OSCD.yaml | 2 +- src/core/trainers.py | 4 ++-- src/train.py | 2 +- 12 files changed, 13 insertions(+), 13 deletions(-) diff --git a/configs/config_EF_AC_Szada.yaml b/configs/config_EF_AC_Szada.yaml index 0c9b917..8579c59 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 91632dd..52a103a 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 9902142..bb1005e 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 f1b5ae9..2e1119b 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 5be5494..d1972fb 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 979facf..1ab90f9 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 640f83d..a007e1e 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 055f970..e569a2b 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 931fc0c..5d6a35d 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 ca28ae3..c871805 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 83330b3..68cde97 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 782336b..c278954 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') -- GitLab