This commit is contained in:
Starbeamrainbowlabs 2022-09-01 18:05:33 +01:00
parent e2f621e212
commit b2f1ba29bb
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -55,7 +55,7 @@ def convnext(
classifier_activation = 'softmax'
# Note that we CAN'T add data_format here, 'cause Dense doesn't support specifying the axis
):
print("convnext:shape IN x", x.shape)
assert len(depths) == len(dims)
def forward_features(x):