acoustic-van-30942
03/26/2023, 11:49 PMfrom fastprogress.fastprogress import master_bar, progress_bar
mb = master_bar(range(1, self.NUMBER_EPOCHS+1))
for epoch in mb:
epoch_loss = 0
for step, batch in enumerate(progress_bar(self.train_dataloader, parent=mb)):
x_batch, y_batch = batch