mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-25 02:32:59 +00:00
debug a
This commit is contained in:
parent
e2f621e212
commit
b2f1ba29bb
1 changed files with 1 additions and 1 deletions
|
@ -55,7 +55,7 @@ def convnext(
|
||||||
classifier_activation = 'softmax'
|
classifier_activation = 'softmax'
|
||||||
# Note that we CAN'T add data_format here, 'cause Dense doesn't support specifying the axis
|
# 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)
|
assert len(depths) == len(dims)
|
||||||
|
|
||||||
def forward_features(x):
|
def forward_features(x):
|
||||||
|
|
Loading…
Reference in a new issue