pbar 고급진 사용법
기술/PyTorch

pbar 고급진 사용법

pbar = tqdm(enumerate(train_dataloader), total=len(train_dataloader))
        model.train()
        for batch, (data, labels) in pbar:
            data, labels = data.to(device), labels.to(device)
            
            # ...
            
            pbar.update()
            pbar.set_description(
                f"Train: [{epoch + 1:03d}] "
                f"Loss: {(epoch_loss / (batch + 1)):.3f}, "
            )
            
        pbar.close()