Error while building graph for 3D ResNet

waleedka / hiddenlayer

Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.

MIT License

1.8k stars 267 forks source link

Error while building graph for 3D ResNet #48

Open anujshah1003 opened 5 years ago

anujshah1003 commented 5 years ago

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

pranavgundewar commented 5 years ago

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]))