Skip to content

GitLab

Commit 100a716c authored by Vishnu Banna's avatar Vishnu Banna
Browse files

comments addrssing

parent b4625e11
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/beta/projects/yolo/configs/yolo.py
View file @ 100a716c
...@@ -342,7 +342,7 @@ def yolo_darknet() -> cfg.ExperimentConfig: ...@@ -342,7 +342,7 @@ def yolo_darknet() -> cfg.ExperimentConfig:
@exp_factory.register_config_factory('scaled_yolo') @exp_factory.register_config_factory('scaled_yolo')
def scaled_yolo() -> cfg.ExperimentConfig: def scaled_yolo() -> cfg.ExperimentConfig:
"""COCO object detection with YOLOv4-csp and v4""" """COCO object detection with YOLOv4-csp and v4"""
train_batch_size = 128 train_batch_size = 64
eval_batch_size = 8 eval_batch_size = 8
train_epochs = 300 train_epochs = 300
warmup_epochs = 3 warmup_epochs = 3
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment