Skip to content

Commit

Permalink
Improve consistency and documentation
Browse files Browse the repository at this point in the history
- Renamed the meta encoder version prefix from "meta" to "m" in convert_coreml_pytorch.py for enhanced consistency.
- Updated CoreML_Backend.md to format the model directory name as code, improving clarity.
  • Loading branch information
ChinChangYang committed Jul 28, 2024
1 parent 825305e commit 322ee23
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion docs/CoreML_Backend.md
Original file line number Diff line number Diff line change
Expand Up @@ -193,4 +193,4 @@ wget https://github.com/lightvector/KataGo/releases/download/v1.15.0/b18c384nbt-
python python/convert_coreml_pytorch.py -checkpoint b18c384nbt-humanv0.ckpt -use-swa
```

This will output the CoreML model directory KataGoModel19x19fp16m1.mlpackage, tailored for the CoreML backend.
This will output the CoreML model directory `KataGoModel19x19fp16m1.mlpackage`, tailored for the CoreML backend.
2 changes: 1 addition & 1 deletion python/convert_coreml_pytorch.py
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ def main():

# Set the meta encoder name
meta_encoder_name = (
"" if meta_encoder_version == 0 else f"meta{meta_encoder_version}"
"" if meta_encoder_version == 0 else f"m{meta_encoder_version}"
)

# Set file name
Expand Down

0 comments on commit 322ee23

Please sign in to comment.