Skip to content

Commit

Permalink
update recipe with reset
Browse files Browse the repository at this point in the history
  • Loading branch information
wonjuleee committed Aug 28, 2024
1 parent 4e1b810 commit 9b0e5f3
Showing 1 changed file with 8 additions and 17 deletions.
25 changes: 8 additions & 17 deletions src/otx/recipe/keypoint_detection/rtmpose_tiny_single_obj.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -27,20 +27,18 @@ engine:

callback_monitor: val/accuracy

data:
task: KEYPOINT_DETECTION
mem_cache_size: 1GB
mem_cache_img_max_size: null
stack_images: true
data_format: coco_person_keypoints
unannotated_items_ratio: 0.0
image_color_channel: RGB
data: ../_base_/data/keypoint_detection.yaml

overrides:
gradient_clip_val: 35.0
reset:
- data.train_subset.transforms
- data.val_subset.transforms
- data.test_subset.transforms
input_size:
- 512
- 512
train_subset:
subset_name: train
batch_size: 32
transforms:
- class_path: otx.core.data.transform_libs.torchvision.TopdownAffine
init_args:
Expand All @@ -56,8 +54,6 @@ data:
mean: [123.675, 116.28, 103.53]
std: [58.395, 57.12, 57.375]
val_subset:
subset_name: val
batch_size: 32
transforms:
- class_path: otx.core.data.transform_libs.torchvision.TopdownAffine
init_args:
Expand All @@ -71,8 +67,6 @@ data:
mean: [123.675, 116.28, 103.53]
std: [58.395, 57.12, 57.375]
test_subset:
subset_name: test
batch_size: 32
transforms:
- class_path: otx.core.data.transform_libs.torchvision.TopdownAffine
init_args:
Expand All @@ -85,6 +79,3 @@ data:
init_args:
mean: [123.675, 116.28, 103.53]
std: [58.395, 57.12, 57.375]

overrides:
gradient_clip_val: 35.0

0 comments on commit 9b0e5f3

Please sign in to comment.