AlexNet_3dConv

TensorFlow implementation of AlexNet(2012) by Alex Krizhevsky, with 3D convolutiional layers.

3D AlexNet

 Network with a standart AlexNet architecture, but it has 3D instead 2D filters on each Conv and Pool layers.

Standart AlexNet

To fit this model into memory, model training proccess can be refactored with @OpenAI's Gradient Checkpointing algorithm https://github.com/openai/gradient-checkpointing

• Note: This model needs a lot of GPU memory. Training session will not starts on 1 GPU without additional data separation or code's parallelization (or https://github.com/openai/gradient-checkpointing for example).

• Note2: This model doesn't pretend to be the SilverBullet in 3D image recognition (like AlexNet was in 2D). It's just an example of 3D convolutional model in TensorFlow.