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.

Sign up for GitHub

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

Jump to bottom

feat: Migrating TFOD to TensorFlow 2.0 #57

Draft
wants to merge 10 commits into
base: master
Choose a base branch
from
Draft

feat: Migrating TFOD to TensorFlow 2.0 #57

wants to merge 10 commits into from

Conversation

javoweb
Copy link
Contributor

@javoweb javoweb commented Jan 12, 2021 •
edited
Loading

The training workflow is working with tensorflow/tensorflow:2.4.0(both CPU and GPU) and Onepanel DL v0.17 images. It has support for tensorboard.

TODOs:

  • Update CVAT for tf2 model inference.
  • Adapt new tf2 models to our needs (Adapt config files to support GPU training and change naming to better integrate with workflows)
  • Upload new supported models to GitHub(or another repository).
  • Add/change hyperparameters/parameters to support new models.

@rushtehrani
Copy link
Contributor

@javoweb can we try using onepanel/dl:0.17.0 image for this and see if it works? The pip installation the support the feature flag we discussed.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@javoweb @rushtehrani