Fixed tqdm import

andikaperbawa · Mar 12, 2022 · bb10b3e · bb10b3e
1 parent 591879b
commit bb10b3e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/labs/notebooks/SaltNet/train.py b/labs/notebooks/SaltNet/train.py
@@ -2,7 +2,7 @@
 import torch
 import torch.nn as nn
 import matplotlib.pyplot as plt
-import tqdm
+from tqdm.notebook import tqdm
 from sklearn.metrics import accuracy_score
 
 from utils import show_tensor_images