Skip to content

Commit c94476f

Browse files
authored
Fix Deim recipe to be able to run bs search (#4664)
1 parent 92e1376 commit c94476f

File tree

3 files changed

+48
-3
lines changed

3 files changed

+48
-3
lines changed

lib/src/otx/recipe/detection/deim_dfine_l.yaml

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -215,7 +215,22 @@ overrides:
215215
train_subset:
216216
batch_size: 8
217217
num_workers: 4
218-
transforms: []
218+
transforms:
219+
- class_path: otx.data.transform_libs.torchvision.Resize
220+
init_args:
221+
scale: $(input_size)
222+
keep_ratio: false
223+
- class_path: otx.data.transform_libs.torchvision.RandomFlip
224+
init_args:
225+
probability: 0.5
226+
- class_path: torchvision.transforms.v2.ToDtype
227+
init_args:
228+
dtype: ${as_torch_dtype:torch.float32}
229+
scale: false
230+
- class_path: torchvision.transforms.v2.Normalize
231+
init_args:
232+
mean: [0.0, 0.0, 0.0]
233+
std: [255.0, 255.0, 255.0]
219234
sampler:
220235
class_path: otx.data.samplers.balanced_sampler.BalancedSampler
221236

lib/src/otx/recipe/detection/deim_dfine_m.yaml

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -214,7 +214,22 @@ overrides:
214214
train_subset:
215215
batch_size: 8
216216
num_workers: 4
217-
transforms: []
217+
transforms:
218+
- class_path: otx.data.transform_libs.torchvision.Resize
219+
init_args:
220+
scale: $(input_size)
221+
keep_ratio: false
222+
- class_path: otx.data.transform_libs.torchvision.RandomFlip
223+
init_args:
224+
probability: 0.5
225+
- class_path: torchvision.transforms.v2.ToDtype
226+
init_args:
227+
dtype: ${as_torch_dtype:torch.float32}
228+
scale: false
229+
- class_path: torchvision.transforms.v2.Normalize
230+
init_args:
231+
mean: [0.0, 0.0, 0.0]
232+
std: [255.0, 255.0, 255.0]
218233
sampler:
219234
class_path: otx.data.samplers.balanced_sampler.BalancedSampler
220235

lib/src/otx/recipe/detection/deim_dfine_x.yaml

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -215,7 +215,22 @@ overrides:
215215
train_subset:
216216
batch_size: 8
217217
num_workers: 4
218-
transforms: []
218+
transforms:
219+
- class_path: otx.data.transform_libs.torchvision.Resize
220+
init_args:
221+
scale: $(input_size)
222+
keep_ratio: false
223+
- class_path: otx.data.transform_libs.torchvision.RandomFlip
224+
init_args:
225+
probability: 0.5
226+
- class_path: torchvision.transforms.v2.ToDtype
227+
init_args:
228+
dtype: ${as_torch_dtype:torch.float32}
229+
scale: false
230+
- class_path: torchvision.transforms.v2.Normalize
231+
init_args:
232+
mean: [0.0, 0.0, 0.0]
233+
std: [255.0, 255.0, 255.0]
219234
sampler:
220235
class_path: otx.data.samplers.balanced_sampler.BalancedSampler
221236

0 commit comments

Comments
 (0)