Skip to content

GitLab

Commit 067c3cd1 authored by Vishnu Banna's avatar Vishnu Banna
Browse files

classification task fix

parent 72257494
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
official/vision/beta/modeling/factory.py
View file @ 067c3cd1
...@@ -51,8 +51,6 @@ def build_classification_model( ...@@ -51,8 +51,6 @@ def build_classification_model(
norm_activation_config=norm_activation_config, norm_activation_config=norm_activation_config,
l2_regularizer=l2_regularizer) l2_regularizer=l2_regularizer)
print(backbone)
model = classification_model.ClassificationModel( model = classification_model.ClassificationModel(
backbone=backbone, backbone=backbone,
num_classes=model_config.num_classes, num_classes=model_config.num_classes,
......
official/vision/beta/projects/yolo/tasks/image_classification.py
View file @ 067c3cd1
...@@ -15,6 +15,7 @@ ...@@ -15,6 +15,7 @@
# ============================================================================== # ==============================================================================
"""Image classification task definition.""" """Image classification task definition."""
from official.core import task_factory from official.core import task_factory
from official.vision.beta.projects.yolo.configs import darknet_classification as exp_cfg from official.vision.beta.projects.yolo.configs import darknet_classification as exp_cfg
......
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