Skip to content

Commit

Permalink
[style] ruff fix
Browse files Browse the repository at this point in the history
  • Loading branch information
hglee98 committed Nov 28, 2024
1 parent 38e02e3 commit d52e698
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion src/netspresso_trainer/models/registry.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,13 @@
from .base import ClassificationModel, DetectionModel, PoseEstimationModel, SegmentationModel, TaskModel
from .full import pidnet
from .heads.classification import fc, fc_conv
from .heads.detection import anchor_decoupled_head, anchor_free_decoupled_head, rtdetr_head, yolo_fastest_head_v2, yolo_detection_head
from .heads.detection import (
anchor_decoupled_head,
anchor_free_decoupled_head,
rtdetr_head,
yolo_detection_head,
yolo_fastest_head_v2,
)
from .heads.pose_estimation import rtmcc
from .heads.segmentation import all_mlp_decoder
from .necks import fpn, lightfpn, rtdetr_hybrid_encoder, yolopafpn
Expand Down

0 comments on commit d52e698

Please sign in to comment.