Changed imports to lightning and updated the overall import versions

InfectionMedicineProteomics / BINN

Generating biologically informed neural networks

MIT License

25 stars 10 forks source link

Closed ErikHartman closed 11 months ago

ErikHartman commented 11 months ago

Switchd frompytorch-lightning to lightning
This required some changes to the device, as it now threw errors for having some tensors on the cpu and some on the gpu.
All tensors should be switched from Tensor to tensor.

ErikHartman commented 11 months ago

This fixes issue #31