使用Linux的一些tips（五）

8、减少验证时的内存

with torch.no_grad():
    for itr, (img, gt_img) in enumerate(dataloader):
              img = img.cuda()
              gt_img = gt_img.cuda()
              one_hot_gt_img = one_hot(gt_img, num_classes=2)
              one_hot_gt_img = one_hot_gt_img.permute(0, 3, 1, 2).contiguous()
              pred, recon = model(img)
              activation = get_activation(pred)
              if opt.use_reconstruction == 1:
                  loss = loss_fn(activation, one_hot_gt_img) + 0.0005 * \
                      reconstruction_loss_fn(
                      recon, (img*opt.data_std + opt.data_mean)/255.0)
              else:
                  loss = loss_fn(activation, one_hot_gt_img)
              iou = calculate_iou(torch.argmax(activation, dim=1), gt_img)
              for k, v in iou.items():
                  if k in ious:
                      ious[k] += v
              losses += loss.item()

9、用su权限操作

 su 
    apt-get update
    apt-get upgrade
    apt-get install sudo
    vim /etc/sudoers
    cike ALL=(ALL:ALL) ALL
    :wq!
    

10、summaryWritter的可视化

 tensorboard --logdir=（log所在的文件夹）

• 如果在服务器上运行，则需要建立服务器和本地的映射

• 然后在浏览器输入127.0.0.1:16006，即可打开Tensorboard面板。

• 1
  

• 2
  

• 3

• 4
  

11、crtl+c也杀不死后台进程的时候

查看进程号ps aux
kill+进程号