def __load_model_weights(self, weight_path):
    print('loading weight file from : {}'.format(weight_path))
    weight = os.path.join(weight_path)
    chkpt = torch.load(weight, map_location=self.__device)
    self.__model.load_state_dict(chkpt)
    print('loading weight file is done')
    (flops, params) = prk.get_flops_params(self.__model.cpu(), (1, 3, 416, 416))
    print(flops, params)
    self.__model = self.__model.cuda()
    del chkpt
def __load_model_weights(self, weight_path):
    print('loading weight file from : {}'.format(weight_path))
    weight = os.path.join(weight_path)
    chkpt = torch.load(weight, map_location=self.__device)
    self.__model.load_state_dict(chkpt)
    print('loading weight file is done')
    (flops, params) = prk.get_flops_params(self.__model.cpu(), (1, 3, 416, 416))
    print(flops, params)
    self.__model = self.__model.cuda()
    del chkpt