()
| 253 | # keeping regular code. |
| 254 | @do_not_convert |
| 255 | def my_custom_policy() -> Policy: |
| 256 | @augmentation |
| 257 | def trace_aug(data, _): |
| 258 | return origin_trace() |
| 259 | |
| 260 | return Policy( |
| 261 | name="SimplePolicy", |
| 262 | num_magnitude_bins=11, |
| 263 | sub_policies=[ |
| 264 | [(trace_aug, 1.0, None), (augmentations.identity, 0.8, None)], |
| 265 | ], |
| 266 | ) |
| 267 | |
| 268 | @pipeline_def(batch_size=2, num_threads=1, device_id=0, enable_conditionals=True) |
| 269 | def pipe(): |