Skip to content

Commit

Permalink
Refactor boolean flags
Browse files Browse the repository at this point in the history
  • Loading branch information
JesusEV committed Aug 28, 2024
1 parent 1881b35 commit 4060cce
Show file tree
Hide file tree
Showing 8 changed files with 14 additions and 11 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -105,7 +105,6 @@

parser = argparse.ArgumentParser()

parser.add_argument("--record_dynamics", action='store_true')
parser.add_argument("--apply_dales_law", type=str.lower, nargs="*", default=[])
parser.add_argument("--batch_size", type=int, default=1)
parser.add_argument("--c_reg", type=float, default=300.0)
Expand All @@ -115,6 +114,7 @@
parser.add_argument("--n_iter", type=int, default=5)
parser.add_argument("--nvp", type=int, default=1)
parser.add_argument("--prevent_weight_sign_change", type=str.lower, nargs="*", default=[])
parser.add_argument('--record_dynamics', action=argparse.BooleanOptionalAction, default=True)
parser.add_argument("--recordings_dir", type=str, default="./")
parser.add_argument("--seed", type=int, default=1)
parser.add_argument("--surrogate_gradient", type=str.lower, default="piecewise_linear")
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -104,8 +104,7 @@
# ~~~~~

parser = argparse.ArgumentParser()
parser.add_argument("--reset_neurons", action='store_true')
parser.add_argument("--record_dynamics", action='store_true')

parser.add_argument("--apply_dales_law", type=str.lower, nargs="*", default=[])
parser.add_argument("--average_gradient", type=bool, default=True)
parser.add_argument("--batch_size", type=int, default=1)
Expand All @@ -115,7 +114,9 @@
parser.add_argument("--n_iter", type=int, default=5)
parser.add_argument("--nvp", type=int, default=1)
parser.add_argument("--prevent_weight_sign_change", type=str.lower, nargs="*", default=[])
parser.add_argument('--record_dynamics', action=argparse.BooleanOptionalAction, default=True)
parser.add_argument("--recordings_dir", type=str, default="./")
parser.add_argument('--reset_neurons', action=argparse.BooleanOptionalAction, default=True)
parser.add_argument("--seed", type=int, default=1)
parser.add_argument("--surrogate_gradient", type=str.lower, default="piecewise_linear")
parser.add_argument("--surrogate_gradient_beta", type=float, default=1.0)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -104,7 +104,6 @@

parser = argparse.ArgumentParser()

parser.add_argument("--record_dynamics", action='store_true')
parser.add_argument("--apply_dales_law", type=str.lower, nargs="*", default=[])
parser.add_argument("--group_size", type=int, default=32)
parser.add_argument("--c_reg", type=float, default=2.0)
Expand All @@ -115,6 +114,7 @@
parser.add_argument("--n_iter_test", type=int, default=1)
parser.add_argument("--nvp", type=int, default=1)
parser.add_argument("--prevent_weight_sign_change", type=str.lower, nargs="*", default=[])
parser.add_argument('--record_dynamics', action=argparse.BooleanOptionalAction, default=True)
parser.add_argument("--recordings_dir", type=str, default="./")
parser.add_argument("--seed", type=int, default=1)
parser.add_argument("--surrogate_gradient", type=str.lower, default="piecewise_linear")
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -104,8 +104,6 @@

parser = argparse.ArgumentParser()

parser.add_argument("--reset_neurons", action='store_true')
parser.add_argument("--record_dynamics", action='store_true')
parser.add_argument("--apply_dales_law", type=str.lower, nargs="*", default=[])
parser.add_argument("--batch_size", type=int, default=2)
parser.add_argument("--group_size", type=int, default=2)
Expand All @@ -117,7 +115,9 @@
parser.add_argument("--n_iter_test", type=int, default=1)
parser.add_argument("--nvp", type=int, default=1)
parser.add_argument("--prevent_weight_sign_change", type=str.lower, nargs="*", default=[])
parser.add_argument('--record_dynamics', action=argparse.BooleanOptionalAction, default=True)
parser.add_argument("--recordings_dir", type=str, default="./")
parser.add_argument('--reset_neurons', action=argparse.BooleanOptionalAction, default=True)
parser.add_argument("--seed", type=int, default=1)
parser.add_argument("--surrogate_gradient", type=str.lower, default="piecewise_linear")
parser.add_argument("--surrogate_gradient_beta", type=float, default=1.0)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -111,12 +111,13 @@
parser.add_argument("--n_iter", type=int, default=5)
parser.add_argument("--nvp", type=int, default=1)
parser.add_argument("--prevent_weight_sign_change", type=str.lower, nargs="*", default=[])
parser.add_argument('--record_dynamics', action=argparse.BooleanOptionalAction, default=True)
parser.add_argument("--recordings_dir", type=str, default="./")
parser.add_argument('--reset_neurons', action=argparse.BooleanOptionalAction, default=True)
parser.add_argument("--seed", type=int, default=1)
parser.add_argument("--surrogate_gradient", type=str.lower, default="piecewise_linear")
parser.add_argument("--surrogate_gradient_beta", type=float, default=1.0)
parser.add_argument("--surrogate_gradient_gamma", type=float, default=0.3)
parser.add_argument("--record_dynamics", type=bool, default=False)

args = parser.parse_args()

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -111,12 +111,13 @@
parser.add_argument("--n_iter", type=int, default=5)
parser.add_argument("--nvp", type=int, default=1)
parser.add_argument("--prevent_weight_sign_change", type=str.lower, nargs="*", default=[])
parser.add_argument('--record_dynamics', action=argparse.BooleanOptionalAction, default=True)
parser.add_argument("--recordings_dir", type=str, default="./")
parser.add_argument('--reset_neurons', action=argparse.BooleanOptionalAction, default=True)
parser.add_argument("--seed", type=int, default=1)
parser.add_argument("--surrogate_gradient", type=str.lower, default="piecewise_linear")
parser.add_argument("--surrogate_gradient_beta", type=float, default=1.0)
parser.add_argument("--surrogate_gradient_gamma", type=float, default=0.3)
parser.add_argument("--record_dynamics", type=bool, default=False)

args = parser.parse_args()

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,6 @@

parser = argparse.ArgumentParser()

parser.add_argument("--record_dynamics", action='store_true')
parser.add_argument("--apply_dales_law", type=str.lower, nargs="*", default=[])
parser.add_argument("--batch_size", type=int, default=1)
parser.add_argument("--c_reg", type=float, default=300.0)
Expand All @@ -111,6 +110,7 @@
parser.add_argument("--n_iter", type=int, default=5)
parser.add_argument("--nvp", type=int, default=1)
parser.add_argument("--prevent_weight_sign_change", type=str.lower, nargs="*", default=[])
parser.add_argument('--record_dynamics', action=argparse.BooleanOptionalAction, default=True)
parser.add_argument("--recordings_dir", type=str, default="./")
parser.add_argument("--seed", type=int, default=1)
parser.add_argument("--surrogate_gradient", type=str.lower, default="piecewise_linear")
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -99,8 +99,6 @@

parser = argparse.ArgumentParser()

parser.add_argument("--reset_neurons", action='store_true')
parser.add_argument("--record_dynamics", action='store_true')
parser.add_argument("--apply_dales_law", type=str.lower, nargs="*", default=[])
parser.add_argument("--average_gradient", type=bool, default=False)
parser.add_argument("--batch_size", type=int, default=1)
Expand All @@ -110,7 +108,9 @@
parser.add_argument("--n_iter", type=int, default=5)
parser.add_argument("--nvp", type=int, default=1)
parser.add_argument("--prevent_weight_sign_change", type=str.lower, nargs="*", default=[])
parser.add_argument('--record_dynamics', action=argparse.BooleanOptionalAction, default=True)
parser.add_argument("--recordings_dir", type=str, default="./")
parser.add_argument('--reset_neurons', action=argparse.BooleanOptionalAction, default=True)
parser.add_argument("--seed", type=int, default=1)
parser.add_argument("--surrogate_gradient", type=str.lower, default="piecewise_linear")
parser.add_argument("--surrogate_gradient_beta", type=float, default=1.0)
Expand Down

0 comments on commit 4060cce

Please sign in to comment.