search

FluxML / FluxTraining.jl

A flexible neural net training library inspired by fast.ai
https://fluxml.ai/FluxTraining.jl
MIT License
118 stars 25 forks source link

Lorenzoh/fix/67 #71

Closed lorenzoh closed 3 years ago

lorenzoh commented 3 years ago

Fixes #67 and a related issue where Metric.device was not being used .