Implement retrain_lds to prepare retrained models for LDS metric calculation. This PR is based on #21. PR #21 got a bit too messy to comment and edit. This PR implements the retraining part. A subsequent PR will implement the metric.
Also see #9 for the initial design.
2. Summary of the change
Implemented retrain_lds in dattri/model_utils/retrain.py
In addition to the initial design, this implementation also added support for training in parallel.
Added unit test in test/dattri/model_utils/test_retrain.py
3. What tests have been added/updated for the change?
[x] Unit test: Typically, this should be included if you implemented a new function/fixed a bug.
Description
This PR implemented
retrain_lds
.1. Motivation and Context
Implement
retrain_lds
to prepare retrained models for LDS metric calculation. This PR is based on #21. PR #21 got a bit too messy to comment and edit. This PR implements the retraining part. A subsequent PR will implement the metric.Also see #9 for the initial design.
2. Summary of the change
retrain_lds
indattri/model_utils/retrain.py
test/dattri/model_utils/test_retrain.py
3. What tests have been added/updated for the change?