Skip to content

Commit

Permalink
add conversion matchers for torchvision builtin models (#508)
Browse files Browse the repository at this point in the history
* add conversion matchers for torchvision builtin models

* update readme: now support torchvision code conversion

* add log in code_consistency_check to get torchvision version in ci

* removed the * marker used to indicate position-only parameters in api_mapping.json

* Modify test cases for built-in models, add ModelAPIBase to compare model outputs
  • Loading branch information
guozixu2001 authored Nov 25, 2024
1 parent 3fa4758 commit 79acc7d
Show file tree
Hide file tree
Showing 38 changed files with 3,129 additions and 1 deletion.
2 changes: 1 addition & 1 deletion README_EN.md
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ The principle is to use Python syntax tree analysis to generate the source code

The conversion logic is static code scanning. Keep the style and structure of the original code unchanged, only convert the Python API, while other Python codes remain unchanged.

Note that only Python native APIs can be converted. For other third-party libraries (such as mmdet, mmcv, torchvision, etc.) encapsulated based on the Pytorch API cannot be converted, and these APIs rely on manual conversion. It is recommended to copy this part of code and then use tools to convert it.
Note that only Python native APIs can be converted. For other third-party libraries (such as mmdet, mmcv, etc.) encapsulated based on the Pytorch API cannot be converted, and these APIs rely on manual conversion. It is recommended to copy this part of code and then use tools to convert it.

The conversion adopts a non-inplace method, convert the original project files one by one to the folder specified by `out_dir`. And the original file will not be changed to facilitate comparison and debugging before and after:

Expand Down
Loading

0 comments on commit 79acc7d

Please sign in to comment.