Add multigpu capability to both train, test and validation

NKI-AI / ahcore

Ahcore is the AI for Oncology core computational pathology toolkit

Apache License 2.0

15 stars 1 forks source link

Closed jonasteuwen closed 5 months ago

jonasteuwen commented 5 months ago

This PR fixes #8 #25, and allows for MultiGPU training and inference. In addition the following changes have been made:

The H5WriterCallback is now based on the new WriterCallback, which is (much) more performant
mypy didn't run for a while, so now it (almost) passes strictly