hi,
thanks!!!
I try train a model of segmentation ,when epoch = 13 Dataloader stucks （The cursor flashes normally and will not throw an error and will not continue.）,The program will stop at this line ‘’ for i, (image, target) in enumerate(tbar):’’ , when epoch < 13 ,such as epoch=11, Can run normally。
self.trainloader = data.DataLoader(trainset, batch_size=args.batch_size,
                                           drop_last=True, shuffle=True, **kwargs)
self.valloader = data.DataLoader(testset, batch_size=args.batch_size,
                                         drop_last=False, shuffle=False, **kwargs)

def training(self, epoch, writer):
        train_loss = 0.0
        self.model.train()
        tbar = tqdm(self.trainloader)
        start_iter = epoch * tbar.total

        for i, (image, target) in enumerate(tbar):
            self.scheduler(self.optimizer, i, epoch, self.best_pred)
            self.optimizer.zero_grad()
           .....................




epoch  = 11

image.png1845×144 24.8 KB

epoch = 13

image.png1846×235 42.6 KB

nvidia-smi show

image.png788×220 30.9 KB