# Init weights, biases | # Init weights, biases | ||||
torch_utils.initialize_weights(self) | torch_utils.initialize_weights(self) | ||||
self._initialize_biases() # only run once | |||||
torch_utils.model_info(self) | |||||
self.info() | |||||
print('') | print('') | ||||
def forward(self, x, augment=False, profile=False): | def forward(self, x, augment=False, profile=False): |