Visual-Behavior / aloception-oss

Aloception is a set of package for computer vision: aloscene, alodataset, alonet.
Other
92 stars 7 forks source link

Detach tensors not used for backprop #299

Closed Aurelien-VB closed 1 year ago

Aurelien-VB commented 1 year ago

Detach tensor of metrics and outputs used for logging.

Maybe the function could be placed somewhere else in the code to be available for other models.