(m)
| 66 | return tree, epoch |
| 67 | |
| 68 | def init_weights_xavier(m): |
| 69 | if type(m) == torch.nn.Conv2d: |
| 70 | torch.nn.init.xavier_normal_(m.weight, gain=torch.nn.init.calculate_gain('sigmoid')) |
| 71 | |
| 72 | def init_weights_kaiming(m): |
| 73 | if type(m) == torch.nn.Conv2d: |
nothing calls this directly
no outgoing calls
no test coverage detected