def logResults(self, epoch_idx, epoch_loss, prefix=''):
    print(f'{prefix} Epoch: {epoch_idx}, Loss: {epoch_loss}')
    if self.wandb:
        self.wandb.log({(prefix + '_loss'): epoch_loss}, step=epoch_idx, commit=False)
def logResults(self, epoch_idx, epoch_loss, prefix=):
    print(f'{prefix} Epoch: {epoch_idx}, Loss: {epoch_loss}')
    if self.wandb:
        self.wandb.log({(prefix + '_loss'): epoch_loss}, step=epoch_idx, commit=False)