Once #378 is done, it would be nice to automatically use it under the hood in the frameworks where we currently support saving the model, i.e.:
https://github.com/iterative/dvclive/blob/41d12d7936c7a781a30da27e7fd12faa00b34a1e/src/dvclive/keras.py#L49-L53
To:
if self.model_file:
if self.save_weights_only:
self.model.save_weights(self.model_file)
else:
self.model.save(self.model_file)
self.live.log_artifact(self.model_file)
Once #378 is done, it would be nice to automatically use it under the hood in the frameworks where we currently support saving the model, i.e.:
https://github.com/iterative/dvclive/blob/41d12d7936c7a781a30da27e7fd12faa00b34a1e/src/dvclive/keras.py#L49-L53
To: