latent-diffusion
latent-diffusion copied to clipboard
pop from empty list
When training my AE, I sometimes output pop from empty list
(like the figure below), which will have an impact on the training.
That seems like a tqdm issue, what's supposed to be written in that spot where you see the error is the iterations per second.
Try commenting out print(e)
in the check_frequency(self, check_idx) function of the ImageLogger class from main.py:
def check_frequency(self, check_idx):
if ((check_idx % self.batch_freq) == 0 or (check_idx in self.log_steps)) and (check_idx > 0 or self.log_first_step):
try:
self.log_steps.pop(0)
except IndexError as e:
# print(e) # pop from empty list
pass
return True
return False