From c3ce40e532c297ea7049c861ead481ac70f3f988 Mon Sep 17 00:00:00 2001 From: Ben Date: Mon, 25 Oct 2021 16:26:11 -0400 Subject: [PATCH 1/3] Argument adding for window size --- main.py | 2 ++ train.py | 6 ++++-- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 9578208..62af0d2 100644 --- a/main.py +++ b/main.py @@ -66,6 +66,8 @@ parser.add_argument('--num_freq_disp', type=int, default=10, dest='num_freq_disp') parser.add_argument('--num_freq_save', type=int, default=50, dest='num_freq_save') +parser.add_argument('--size_window', type=tuple, default=(5, 5), dest='size_window') + PARSER = Parser(parser) def main(): diff --git a/train.py b/train.py index 78c34c0..86637cc 100644 --- a/train.py +++ b/train.py @@ -51,6 +51,8 @@ def __init__(self, args): self.num_freq_disp = args.num_freq_disp self.num_freq_save = args.num_freq_save + self.size_window = args.size_window + self.gpu_ids = args.gpu_ids if self.gpu_ids and torch.cuda.is_available(): @@ -114,7 +116,7 @@ def train(self): nch_ker = self.nch_ker size_data = (self.ny_in, self.nx_in, self.nch_in) - size_window = (5, 5) + size_window = self.size_window norm = self.norm name_data = self.name_data @@ -329,7 +331,7 @@ def test(self): nch_ker = self.nch_ker size_data = (self.ny_in, self.nx_in, self.nch_in) - size_window = (5, 5) + size_window = self.size_window norm = self.norm From 45cc5647609506321e6d943125e95403137d9360 Mon Sep 17 00:00:00 2001 From: Ben Date: Mon, 25 Oct 2021 16:35:11 -0400 Subject: [PATCH 2/3] Tuple as input did not work, splitting them up --- main.py | 4 +++- train.py | 9 +++++---- 2 files changed, 8 insertions(+), 5 deletions(-) diff --git a/main.py b/main.py index 62af0d2..12e39a7 100644 --- a/main.py +++ b/main.py @@ -66,7 +66,9 @@ parser.add_argument('--num_freq_disp', type=int, default=10, dest='num_freq_disp') parser.add_argument('--num_freq_save', type=int, default=50, dest='num_freq_save') -parser.add_argument('--size_window', type=tuple, default=(5, 5), dest='size_window') +parser.add_argument('--size_window_x', type=int, default=5, dest='size_window_x') + +parser.add_argument('--size_window_y', type=int, default=5, dest='size_window_y') PARSER = Parser(parser) diff --git a/train.py b/train.py index 86637cc..5572435 100644 --- a/train.py +++ b/train.py @@ -51,8 +51,9 @@ def __init__(self, args): self.num_freq_disp = args.num_freq_disp self.num_freq_save = args.num_freq_save - self.size_window = args.size_window - + self.size_window_x = args.size_window_x + self.size_window_y = args.size_window_y + self.gpu_ids = args.gpu_ids if self.gpu_ids and torch.cuda.is_available(): @@ -116,7 +117,7 @@ def train(self): nch_ker = self.nch_ker size_data = (self.ny_in, self.nx_in, self.nch_in) - size_window = self.size_window + size_window = tuple(self.size_window_x, self.size_window_y) norm = self.norm name_data = self.name_data @@ -331,7 +332,7 @@ def test(self): nch_ker = self.nch_ker size_data = (self.ny_in, self.nx_in, self.nch_in) - size_window = self.size_window + size_window = tuple(self.size_window_x, self.size_window_y) norm = self.norm From f3e63888dcac325b6d4ba6222f9ffba38224d6c8 Mon Sep 17 00:00:00 2001 From: Ben Date: Mon, 25 Oct 2021 16:36:56 -0400 Subject: [PATCH 3/3] Python syntax.. ugh --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 5572435..10e5894 100644 --- a/train.py +++ b/train.py @@ -117,7 +117,7 @@ def train(self): nch_ker = self.nch_ker size_data = (self.ny_in, self.nx_in, self.nch_in) - size_window = tuple(self.size_window_x, self.size_window_y) + size_window = (self.size_window_x, self.size_window_y) norm = self.norm name_data = self.name_data @@ -332,7 +332,7 @@ def test(self): nch_ker = self.nch_ker size_data = (self.ny_in, self.nx_in, self.nch_in) - size_window = tuple(self.size_window_x, self.size_window_y) + size_window = (self.size_window_x, self.size_window_y) norm = self.norm