Constructs a CNN model. Args: pretrained (bool): If True, returns a pre-trained model. Returns: The created CNN model.
(pretrained=False, **kwargs)
| 65 | |
| 66 | |
| 67 | def create_model(pretrained=False, **kwargs): |
| 68 | """Constructs a CNN model. |
| 69 | Args: |
| 70 | pretrained (bool): If True, returns a pre-trained model. |
| 71 | |
| 72 | Returns: |
| 73 | The created CNN model. |
| 74 | """ |
| 75 | model = MLP(**kwargs) |
| 76 | |
| 77 | return model |
| 78 | |
| 79 | |
| 80 | __all__ = ["MLP", "create_model"] |