Skip to content

Commit fa1bb57

Browse files
authored
models: fix drn-d-38 conversion for py3.7 and above (#3179)
1 parent ccb3427 commit fa1bb57

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

models/public/drn-d-38/model.yml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,10 @@ postprocessing:
4747
file: models/segment.py
4848
pattern: 'import data_transforms as transforms'
4949
replacement: '# \g<0>'
50+
- $type: regex_replace
51+
file: models/segment.py
52+
pattern: ' target = target.cuda\(async=True\)'
53+
replacement: '# \g<0>'
5054
conversion_to_onnx_args:
5155
- --model-path=$config_dir
5256
- --model-path=$dl_dir/models

0 commit comments

Comments
 (0)