fix (#172): support cpu in knight example

BiomedSciAI / fuse-med-ml

A python framework accelerating ML based discovery in the medical field by encouraging code reuse. Batteries included :)

Apache License 2.0

134 stars 34 forks source link

Closed agiova closed 2 years ago

agiova commented 2 years ago

Adding conditional configuration of lightning trainer depending on cpu or gpu, fixes issue #172