Error in val_loop

[AndreaBrg]

Hi, I'm encountering the following error in val_loop:

Exception in thread Thread-3:
Traceback (most recent call last):
  File "/usr/lib/python3.10/threading.py", line 1016, in _bootstrap_inner
    self.run()
  File "/usr/local/lib/python3.10/dist-packages/ffcv/loader/epoch_iterator.py", line 84, in run
    result = self.run_pipeline(b_ix, ixes, slot, events[slot])
  File "/usr/local/lib/python3.10/dist-packages/ffcv/loader/epoch_iterator.py", line 146, in run_pipeline
    results = stage_code(**args)
  File "", line 2, in stage_code_1
  File "/usr/local/lib/python3.10/dist-packages/ffcv/transforms/ops.py", line 56, in to_device
    dst.copy_(inp, non_blocking=self.non_blocking)
RuntimeError: Inplace update to inference tensor outside InferenceMode is not allowed.You can make a clone to get a normal tensor before doing inplace update.See https://github.com/pytorch/rfcs/pull/17 for more details.

The training loop, works fine.

Any idea of the causes and how to solve it?