From 2497e32a44bfd115ec97dae9704479108421e553 Mon Sep 17 00:00:00 2001 From: Nicola Tonellotto Date: Thu, 28 May 2020 14:44:24 +0200 Subject: [PATCH] Update train.py --- cedr/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cedr/train.py b/cedr/train.py index e0363cf..ff3c63c 100644 --- a/cedr/train.py +++ b/cedr/train.py @@ -3,7 +3,7 @@ import subprocess import random import tempfile -from tqdm import tqdm +from tqdm.autonotebook import tqdm import torch from . import modeling from . import data