Imported from https://github.com/ptmcgrat/3D-ResNets-PyTorch