Skip to content

Should detach the Tensor before decollate_batch #2369

Description

@Nic-Ma

Describe the bug
Currently, we decollate a Tensor data(like: pred) from batch to list without detach() first.
I will enhance it soon.

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Type

No type
No fields configured for issues without a type.

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions