Pytorch_CNN_mixt_representation