Is there a way can I visualize the complete training loop for the GAN architecture in TensorBoard using Pytorch? I think it\'s possible using TF, but I am having a hard time