Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

cog train: bring functionality on par with cog predict #1391

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

yorickvP
Copy link
Contributor

  • Currently, cog train always writes to weights
  • Currently, there's no way to use it with already existing images
  • Some models have a weights directory

Together, this makes for a terrible experience. This PR adds a way to specify the output directory and use an existing image.

Signed-off-by: Yorick van Pelt <yorick@yorickvanpelt.nl>
Signed-off-by: Yorick van Pelt <yorick@yorickvanpelt.nl>
@anotherjesse
Copy link
Contributor

@yorickvP I haven't run this - but the weights are/were base64 datauris inside a json file!

this is extremely painful

@yorickvP yorickvP requested a review from mattt February 8, 2024 15:10
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants