Sampling after training

Hi there,

Firstly,  Apologizes if my question does not make so much sense. I tried to run this code in my local environment and started with the main.py but I could not figure out how to sample data after this training step. I am new with deep learning models.

Thank you so much