error raised after test_epoch_end()

[teang1995]

Hi there,The error below was obtained when I run trainer.test().
I've got result from trainer.test_epoch_end() successfully, but after that error AttributeError: 'int' object has no attribute 'cpu' occured.
Here is full log i got after runed trainer.test.
I run code at Colab, I found that error not obtained without using gpu.
I cannt figure out reason.
I'd really appreciate your help.

GPU available: True, used: True
TPU available: False, using: 0 TPU cores
Reusing dataset story (/root/.cache/huggingface/datasets/story/example/1.0.0/4f914336e4df642ed7f022f2b7554d8405f25a1ecb77125af9bcb42a96fb2311)
LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0]
Testing: 0it [00:00, ?it/s]
batch:0 (0.0, 0.159375, 0.8999999761581421, 1.0, 1.0)

batch:1 (0.0, 0.3193750000000001, 0.7250000238418579, 1.0, 1.0)

batch:2 (0.0, 1.1163888888888887, 0.75, 1.0, 1.0)

batch:3 (0.0, 1.0423437500000001, 0.800000011920929, 1.0, 1.0)

batch:4 (0.0, 0.8610999999999999, 0.8399999737739563, 1.0, 1.0)

batch:5 (0.0, 1.1055555555555552, 0.8583333492279053, 1.0, 1.0)

batch:6 (0.0, 5.589100346020762, 0.7573529481887817, 1.0, 1.0)
Testing: 100% 7/7 [00:04<00:00,  1.63it/s]
### final ###
median: 0.0
var: 5.589100346020762
top1: 0.7573529481887817
top10: 1.0
top100: 1.0
Traceback (most recent call last):
  File "/usr/lib/python3.7/runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "/usr/lib/python3.7/runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "/content/wisdomify/wisdomify/main/eval.py", line 60, in <module>
    main()
  File "/content/wisdomify/wisdomify/main/eval.py", line 56, in main
    trainer.test(model=wisdomifier.rd, datamodule=data_module, verbose=False)
  File "/usr/local/lib/python3.7/dist-packages/pytorch_lightning/trainer/trainer.py", line 581, in test
    results = self._run(model)
  File "/usr/local/lib/python3.7/dist-packages/pytorch_lightning/trainer/trainer.py", line 761, in _run
    self.post_dispatch()
  File "/usr/local/lib/python3.7/dist-packages/pytorch_lightning/trainer/trainer.py", line 791, in post_dispatch
    self.accelerator.teardown()
  File "/usr/local/lib/python3.7/dist-packages/pytorch_lightning/accelerators/gpu.py", line 51, in teardown
    self.lightning_module.cpu()
  File "/usr/local/lib/python3.7/dist-packages/pytorch_lightning/utilities/device_dtype_mixin.py", line 135, in cpu
    return super().cpu()
  File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 667, in cpu
    return self._apply(lambda t: t.cpu())
  File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 530, in _apply
    module._apply(fn)
  File "/usr/local/lib/python3.7/dist-packages/torchmetrics/metric.py", line 317, in _apply
    setattr(this, key, [fn(cur_v) for cur_v in current_val])
  File "/usr/local/lib/python3.7/dist-packages/torchmetrics/metric.py", line 317, in <listcomp>
    setattr(this, key, [fn(cur_v) for cur_v in current_val])
  File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 667, in <lambda>
    return self._apply(lambda t: t.cpu())
AttributeError: 'int' object has no attribute 'cpu'

[rohitgr7]

can you share a minimal example to reproduce this issue?
https://colab.research.google.com/drive/1HvWVVTK8j2Nj52qU4Q4YCyzOm0_aLQF3?usp=sharing#scrollTo=4Dk6Ykv8lI7X