no save_weights function about pytorch-yolov3 HOT 4 CLOSED

def save_weights(self, path, cutoff=-1):

    if cutoff:
        num_layers = len(self.module_list)
    else:
        num_layers = cutoff

    fp = open(path, 'wb')
    self.header_info[3] = self.seen
    self.header_info.tofile(fp)

    for i, (module_def, module) in enumerate(zip(self.module_defs[:cutoff], self.module_list[:cutoff])):
        if module_def['type'] == 'convolutional':
            conv_layer  = module[0]

            # If batch norm, load bn first
            if module_def['batch_normalize']:
                bn_layer    = module[1]

                bn_layer.bias.data.cpu().numpy().tofile(fp)
                bn_layer.weight.data.cpu().numpy().tofile(fp)
                bn_layer.running_mean.data.cpu().numpy().tofile(fp)
                bn_layer.running_var.data.cpu().numpy().tofile(fp)
            # Load conv bias
            else:
                conv_layer.bias.data.cpu().numpy().tofile(fp)

            # Load conv weights
            conv_layer.weight.data.cpu().numpy().tofile(fp)

    fp.close()