Basic CNN ##. More...
Inheritance diagram for model.LeNet:
Collaboration diagram for model.LeNet:
Public Member Functions | |
| def | __init__ (self, num_inp, num_out) |
| def | forward (self, x) |
| def | __str__ (self) |
Public Attributes | |
| conv1 | |
| pool | |
| conv2 | |
| pool2 | |
| fc1 | |
| fc2 | |
Detailed Description
Basic CNN ##.
Basic CNN.
Constructor & Destructor Documentation
◆ __init__()
| def model.LeNet.__init__ | ( | self, | |
| num_inp, | |||
| num_out | |||
| ) |
Init LeNet.
Member Function Documentation
◆ __str__()
| def model.LeNet.__str__ | ( | self | ) |
Get name of model
◆ forward()
| def model.LeNet.forward | ( | self, | |
| x | |||
| ) |
Main method of LeNet
The documentation for this class was generated from the following file:
- /home/antoineh/github/smart_augmentation/higher/smart_aug/model.py
