分布式debug方式 合集 持续更新

test1：检查dataset类方法

from zqy_cfpns_cov_mulc_75_load import TamperDataset_train
from torch.utils.data import Dataset, DataLoader
import torch
import argparse
import tempfile
# zqy
parser = argparse.ArgumentParser()
parser.add_argument('--imgs', type=str, default='image')
parser.add_argument('--masks', type=str, default='mask')
parser.add_argument('--pilt', type=int, default=0)
parser.add_argument('--numw', type=int, default=0)
args = parser.parse_args()
use_pilt = (args.pilt == 1)
train_data3 = TamperDataset_train(args.imgs,args.masks,False,0, pilt=use_pilt)
# train_sampler1 = torch.utils.data.distributed.DistributedSampler(train_data3, shuffle=False)

train_loader1 = DataLoader(dataset=train_data3, batch_size=6, num_workers=args.numw, shuffle=False)

for i in train_loader1:
    print(i) 

小tips：找到报错行，在 console中输入，依次删去，找到具体报错变量。