Skip to content

feat: Migrating TFOD to TensorFlow 2.0#57

Draft
javoweb wants to merge 10 commits intoonepanelio:masterfrom
javoweb:feat/tfod/tf2
Draft

feat: Migrating TFOD to TensorFlow 2.0#57
javoweb wants to merge 10 commits intoonepanelio:masterfrom
javoweb:feat/tfod/tf2

Conversation

@javoweb
Copy link
Contributor

@javoweb javoweb commented Jan 12, 2021

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants