Should detach the Tensor before decollate_batch #2369
Description
Describe the bug
Currently, we decollate a Tensor data(like: pred) from batch to list without detach() first.
I will enhance it soon.
{{ message }}
Describe the bug
Currently, we decollate a Tensor data(like: pred) from batch to list without detach() first.
I will enhance it soon.