我在代码中使用了PyTorch DataLoader,并且存在循环使用此Dataloader中的数据。
但是,当我尝试结束此循环时,它失败了。
没有错误信息。该程序仍在。
GPU不是内存不足,但易失性GPU-util始终保持为0%
def function():
seq_dt = SequenceLoader(image_list)
seq_loader = DataLoader(seq_dt, batch_size=1, shuffle=False, sampler=None,
num_workers=1, pin_memory=True)
for image_idx, img in enumerate(seq_loader):
...
...
my_variable = ...
...
if my_variable<my_threshold:
return my_flag
能帮我解决这个问题吗?
谢谢