Open jdeschamps opened 5 days ago
Can’t take a proper look rn but the last paragraph of the docstring in CAREamist.__init__
mentions the experiment name so I guess this should also be removed.
Fixes #195
Can’t take a proper look rn but the last paragraph of the docstring in CAREamist.init mentions the experiment name so I guess this should also be removed.
Good catch!
Description
If CAREamics is used in a multithreaded application (e.g. napari plugin), then we need a way to stop the training programmatically. This PR add a
stop_training
method to theCAREamist
.Additionally, this PR removes the unused parameter
experiment_name
from the CAREamist.stop_training
method that raises theshould_stop
flag of Lightning.Changes Made
careamist.py
, by adding thestop_training
function and cleaning up unused keywords.Please ensure your PR meets the following requirements: