PyTorch编程时的小技巧

目录

1.训练时显示进度条

2.终止jupyter cell正在运行的代码

3.查看服务器GPU的使用情况

---

1.训练时显示进度条

# pip install tqdm
from tqdm import tqdm  # 进度条

对加载器循环时，写如下代码

 for data,target in tqdm(train_iter, total = len(train_iter), position = 0, leave = True):

或

for idx,batch in tqdm(enumerate(test_iter), total = len(test_iter), position = 0, leave = True):
                data = batch[0].to(device)
                target = batch[1].to(device)
                outputs = model(data)

 40000代表总批数

2.终止jupyter cell正在运行的代码

鼠标点在对应的cell内，再按Esc键，然后再按两次键盘英文字母'I'键，即可终止

3.查看服务器GPU的使用情况

nvidia-smi