binzhangbin

This net need fixed size image as input?

15

Traceback (most recent call last): File "train.py", line 140, in output = model(X_batch) File "/home/zzp/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1051, in _call_impl return forward_call(*input, **kwargs) File "/home/zzp/2t/binzhang/Medical-Transformer/lib/models/axialnet.py", line 507, in forward return self._forward_impl(x)...