search

connorlee77 / pytorch-mutual-information

Mutual Information in Pytorch
103 stars 10 forks source link

fix unreferenced device bug #5

Closed connorlee77 closed 1 year ago

connorlee77 commented 1 year ago

Fix the bug referenced in #4