Skip to content

Commit

Permalink
write a new model file new edition
Browse files Browse the repository at this point in the history
  • Loading branch information
chenyang23333 committed Jul 11, 2023
1 parent f4feec2 commit 7237278
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions docs/zh/how_to_guides/write_a_new_model.md
Original file line number Diff line number Diff line change
Expand Up @@ -136,6 +136,8 @@ default_cfgs = {"mymodel": _cfg(url="")}
- 在`construct`函数中定义模型前向逻辑。 <br>

```python
class MYmodel(nn.Cell):
def __init__(self, cfg, in_channels=3, num_classes=None, sync_bn=False):
super(MYmodel, self).__init__()
Expand Down

0 comments on commit 7237278

Please sign in to comment.