Skip to content

Commit 220b3c8

Browse files
committed
Fix MLP model
1 parent 99e9bea commit 220b3c8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def main():
128128
# You can play around with any model here
129129
# The chosen setup will append a final layer so that the output is mu, sigma, and weights
130130
model = None
131-
if args.model == 'MLP':
131+
if args.model == 'mlp':
132132
model = MLP(args.hidden_layers, args.activation, args.resnet)
133133

134134
setup = qsetup.construct(system, model, xi, A, B, args)

0 commit comments

Comments
 (0)