Commit 35e9d291 authored by anivegesana's avatar anivegesana
Browse files

Remove debugging registrations

parent 0bfc786c
......@@ -64,7 +64,6 @@ class RevNet(hyperparams.Config):
# Specifies the depth of RevNet.
model_id: int = 56
from official.vision.beta.projects.yolo.configs.backbones import DarkNet
@dataclasses.dataclass
class Backbone(hyperparams.OneOfConfig):
......@@ -86,4 +85,3 @@ class Backbone(hyperparams.OneOfConfig):
efficientnet: EfficientNet = EfficientNet()
spinenet: SpineNet = SpineNet()
mobilenet: MobileNet = MobileNet()
darknet: DarkNet = DarkNet()
......@@ -48,6 +48,7 @@ class Parser(parser.Parser):
aug_rand_hflip=True,
dtype='float32'):
"""Initializes parameters for parsing annotations in the dataset.
Args:
output_size: `Tenssor` or `list` for [height, width] of output image. The
output_size should be divided by the largest feature stride 2^max_level.
......
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