search

vishnu-m77 / DiffAPTOSC

Final Project for AMATH 495
1 stars 2 forks source link

Why is .cpu() used here? #30

Closed vishnu-m77 closed 11 months ago

vishnu-m77 commented 11 months ago

https://github.com/vishnu-m77/DiffMIC/blame/7e0bb07b39cd902ba6992d7a2b02c316e90fb93e/src/metrics.py#L31

Lakshay-G commented 11 months ago

Not sure. I didn't implement compute_f1_score().

Sehmimul commented 11 months ago

cpu is used to convert torch tensor to numpy ndarray object

vishnu-m77 commented 11 months ago

Code works without .cpu()

Sehmimul commented 11 months ago

okay