Se han modificado 1 ficheros con 0 adiciones y 1 borrados
+ 0
- 1
code/resnet18/resnet18.py
Ver fichero
| self.__strides = [2, 1] if down_sample else [1, 1] | self.__strides = [2, 1] if down_sample else [1, 1] | ||||
| KERNEL_SIZE = (3, 3) | KERNEL_SIZE = (3, 3) | ||||
| # use He initialization, instead of Xavier (a.k.a 'glorot_uniform' in Keras), as suggested in [2] | |||||
| INIT_SCHEME = "he_normal" | INIT_SCHEME = "he_normal" | ||||
| self.conv_1 = Conv2D(self.__channels, strides=self.__strides[0], | self.conv_1 = Conv2D(self.__channels, strides=self.__strides[0], |
Cargando…