Xylambda/torchfitter

BUG: metrics are not being moved to execution device

Closed this issue · 0 comments

Xylambda commented

Passed metrics need to be moved to whatever device the Accelerator object finds