Fix cpu().detach() into detach().cpu()

facebookresearch / d2go

D2Go is a toolkit for efficient deep learning

Apache License 2.0

838 stars 201 forks source link

Closed DrYangLiu closed 1 year ago

DrYangLiu commented 1 year ago

Summary: For visualisation, tensor variables should be detached from the computational graph. The .cpu() function call should be after the detach().

Reviewed By: frabu6

Differential Revision: D48737228

facebook-github-bot commented 1 year ago

This pull request was exported from Phabricator. Differential Revision: D48737228

facebook-github-bot commented 1 year ago

This pull request was exported from Phabricator. Differential Revision: D48737228

facebook-github-bot commented 1 year ago

This pull request has been merged in facebookresearch/d2go@93037c4e91d35a732823e5eb2c9886eeabf4e690.