Skip to content

frameworks: Use log_artifact #465

@daavoo

Description

@daavoo

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)

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions