Skip to content

Commit 6bf40ba

Browse files
author
liubigli
committed
added conf files for each task: classification, segmentation, object detection, panoptic, registration
1 parent 259a0f2 commit 6bf40ba

6 files changed

+45
-0
lines changed

conf/task/classification.yaml

+7
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
# @package task
2+
defaults:
3+
- /task/default
4+
- override /data@_group_: classification/default
5+
- override /models@_group_: segmentation/default
6+
7+
name: classification

conf/task/default.yaml

+10
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
# @package task
2+
defaults:
3+
- /data@_group_: default
4+
- /models@_group_: default
5+
6+
# By default.yaml we turn off recursive instantiation, allowing the user to instantiate themselves at the appropriate times.
7+
_recursive_: false
8+
9+
#_target_: lightning_transformers.core.model.TaskTransformer
10+
lr_scheduler: ${lr_scheduler}

conf/task/object_detection.yaml

+7
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
# @package task
2+
defaults:
3+
- /task/default
4+
- override /data@_group_: object_detection/default
5+
- override /models@_group_: object_detection/default
6+
7+
name: object_detection

conf/task/panoptic.yaml

+7
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
# @package task
2+
defaults:
3+
- /task/default
4+
- override /data@_group_: panoptic/default
5+
- override /models@_group_: panoptic/default
6+
7+
name: panoptic

conf/task/registration.yaml

+7
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
# @package task
2+
defaults:
3+
- /task/default
4+
- override /data@_group_: registration/default
5+
- override /models@_group_: registration/default
6+
7+
name: registration

conf/task/segmentation.yaml

+7
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
# @package task
2+
defaults:
3+
- /task/default
4+
- override /data@_group_: segmentation/default
5+
- override /models@_group_: segmentation/default
6+
7+
name: segmentation

0 commit comments

Comments
 (0)