Как отключить прогресс бар в google colab при тренировке нейронной сети

Тренирую нн в гугл колабе:

model = nn.Sequential(
    ...
)

optimizer = torch.optim.SGD(model.parameters(), lr=0.05)


for epoch in range(10):
    for x_train, y_train in tqdm(train_dataloader): 
        y_pred = model(x_train) 
        loss = F.cross_entropy(y_pred, y_train)  
        loss.backward()
        optimizer.step() 
        optimizer.zero_grad() 

    val_accuracy = []
    with torch.no_grad():
        for x_val, y_val in tqdm(val_dataloader):
            y_pred = model(x_val) 
            val_accuracy.extend(
                (torch.argmax(y_pred, dim=-1) == y_val).numpy().tolist()
            )

При при этом появляются десятки загрузочных полос:

Как отключить данную функцию?