Skip to content

Commit 262ff1a

Browse files
committed
removed default values for neurons and act funcs
1 parent 304d5f1 commit 262ff1a

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

algorithms/optimize.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,8 @@
1515
parser.add_argument("--dim", type=int, help="number of dimensions")
1616
parser.add_argument("--n_init", type=int, help="initial number of samples")
1717
parser.add_argument("--max_evals", type=int, help="maximum number of function evaluations")
18-
parser.add_argument("--neurons", nargs='+', type=int, default=[256, 256],
19-
help="list of int representing neurons in each layer, only for EBONN")
20-
parser.add_argument("--act_funcs", nargs='+', type=str, default=["GELU", "GELU"],
21-
help="list of str representing activation function in each layer, only for EBONN")
18+
parser.add_argument("--neurons", nargs='+', type=int, help="list of int representing neurons in each layer, only for EBONN")
19+
parser.add_argument("--act_funcs", nargs='+', type=str, help="list of str representing activation function in each layer, only for EBONN")
2220
parser.add_argument("--seed", type=int, default=None, help="seed for the run")
2321

2422
args = parser.parse_args()

0 commit comments

Comments
 (0)