From 2d8622f224ca90ace6b7011ddd691d2d44573deb Mon Sep 17 00:00:00 2001 From: Erin Grant Date: Sat, 2 Dec 2023 20:45:59 +0000 Subject: [PATCH] tuple --- nets/models/feedforward.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nets/models/feedforward.py b/nets/models/feedforward.py index 9b5b881..6dcefe8 100644 --- a/nets/models/feedforward.py +++ b/nets/models/feedforward.py @@ -108,7 +108,7 @@ def __init__( for key, ins, outs, drop in zip( hidden_keys, - (in_features,) + hidden_features[:-1], + (in_features, *tuple(hidden_features[:-1])), hidden_features, dropout_probs[:-1], strict=True,