hiddenlayer icon indicating copy to clipboard operation
hiddenlayer copied to clipboard
verified publisher icon waleedka

Error while building graph for 3D ResNet

Open anujshah1003 opened this issue 6 years ago • 1 comments

I am getting this error while building graph for 3D ResNet: hl.build_graph(model,torch.zeros([1,3,16,112,12]))

RuntimeError: invalid argument 2: input image (T: 1 H: 4 W: 1) smaller than kernel size (kT: 1 kH: 4 kW: 4) at /opt/conda/conda-bld/pytorch_1549635019666/work/aten/src/THCUNN/generic/VolumetricAveragePooling.cu:57

anujshah1003 avatar Jul 15 '19 05:07 anujshah1003

I believe your input dimensions for visualization might be wrong. Input to the model should be hl.build_graph(model,torch.zeros([1,3,16,112,112]))

pranavgundewar avatar Sep 17 '19 16:09 pranavgundewar