Есть ли возможный способ разорвать цикл загрузки данных?

Question

Я использую PyTorch DataLoader в своем коде, и есть цикл, использующий данные из этого загрузчика данных.
Однако, когда я пытаюсь завершить этот цикл, это не удалось. Там нет информации об ошибке. Программа до сих пор. GPU не хватает памяти, но энергозависимая утилита GPU сохраняет значение 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

Не могли бы вы помочь мне с этой проблемой?
Спасибо