tensorflow / tensorflow/models
Cloud Training EfficientDet (d4)
@pkulzc is already working on this.
Since Sep 1, 2020.
- Dominant language
- Python
- Stars
- 77.7k
- Forks
- 44.8k
- PR merge metrics
- No merged PRs in 30d
Description
I've tried to train a efficientdet-d4 model, based on a config. Because configs for efficientdets and tf2 are just for TPUs I tried this option, right starting ckpt (D4) was used from the model zoo.
# From the tensorflow/models/research/ directory
cp object_detection/packages/tf2/setup.py .
gcloud ai-platform jobs submit training whoami_object_detection_date +%m_%d_%Y_%H_%M_%S \
--job-dir=gs://${MODEL_DIR} \
--package-path ./object_detection \
--module-name object_detection.model_main_tf2 \
--runtime-version 2.1 \
--python-version 3.6 \
--scale-tier BASIC_TPU \
--region us-central1 \
-- \
--use_tpu true \
--model_dir=gs://${MODEL_DIR} \
--pipeline_config_path=gs://${PIPELINE_CONFIG_PATH}
This example from the docs .md throws several error before the last one:
- python version should be 3.7
- problem with accessing a restricted bucket (it seams as an open issue regarding tpu cloud train...)
Current obstacle is:

Does anybody experienced something similar?
Thanks,
Alex
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.