有没有可能打破数据加载器循环的方法?

时间:2019-04-24 09:09:26

标签: gpu pytorch

我在代码中使用了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

能帮我解决这个问题吗?
谢谢

0 个答案:

没有答案