Make plot model structure optional

DeepRegNet / DeepReg

Medical image registration using deep learning

Apache License 2.0

564 stars 76 forks source link

Make plot model structure optional #685

Closed mathpluscode closed 3 years ago

mathpluscode commented 3 years ago

Subject of the issue

With #656, we are now saving model structure while training. However, it requires the graphviz package which can not be installed via pip. Users may experience bugs in case the package was not installed.

Therefore we need to

Update the docs
Make this optional, in case of ImportError we print a warning and skip.