Implement a 3D custom model

This post has the output size calculation from the docs written as a Python function. You can enter in the arguments on one dimension for your Conv3d and MaxPool3d as well as input size on one dimension to determine what the output size is you should expect from that layer, given a certain input size.