docstr: test load_state_dict_with_same_shape
docstr_tok: ["test","load_state_dict_with_same_shape"]
code: def test_load_pretrained_model(self):
      """
      test load_state_dict_with_same_shape
      """
      model1 = MockModel()
      model2 = MockModel_()
  
      w1 = model1.state_dict()
  
      model2.load_state_dict_with_same_shape(w1)
      w2 = model2.state_dict()
      for k, p in w2.items():
          if "nn.2." not in k:
              torch.testing.assert_allclose(w1[k], p)
  
imports: import unittest
  import torch
  import omegaconf
  from omegaconf import OmegaConf, DictConfig
  from torch.nn import Sequential, Linear, ReLU, LeakyReLU, BatchNorm1d, Dropout
  import os
  import sys
  from torch_geometric.data import Data
  from torch_points3d.models.base_model import BaseModel
  from test.mockdatasets import MockDatasetGeometric
  from torch_points3d.models.model_factory import instantiate_model
label_desc: Use image node_id 3 for calling the TestBaseModel obj's underlying member method code with example usage: obj.test_load_pretrained_model() without return types