What is the definition of [`train_writer`](https://github.com/denti/AlexNet3D/blob/5c84a2785ace5b8c3ae1af6369b99564bda9a9ad/AlexNet_3dConv.py#L160)?