implement https://arxiv.org/abs/1602.04105 using pytorch follow https://github.com/radioML/examples/blob/master/modulation_recognition/RML2016.10a_VTCNN2_example.ipynb