Thanks for your code for the project! I need to find out the log_sample() after the batch training. Is it used to reconstruct the sample at each end of the training epoch? And then, put the reconstructed samples into checkpoint/name/sample folder?