Skip to content

Commit f424e5e

Browse files
authored
Merge pull request #826 from rushtehrani/fix/tfod
fix: Rename train.py => main.py in TFOD workflow template
2 parents fbc780a + d9cc564 commit f424e5e

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

db/yaml/workflows/tf-object-detection-training/20201223202929.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ templates:
101101
cd /mnt/src/tf/research/ && \
102102
/mnt/src/protoc/bin/protoc object_detection/protos/*.proto --python_out=. && \
103103
cd /mnt/src/train/workflows/tf-object-detection-training && \
104-
python train.py \
104+
python main.py \
105105
--extras="{{workflow.parameters.hyperparameters}}" \
106106
--model="{{workflow.parameters.cvat-model}}" \
107107
--num_classes="{{workflow.parameters.cvat-num-classes}}" \

0 commit comments

Comments
 (0)