Loading issue in trained model

[arrrrr3186]

python3 test.py --dataroot ./Custom_data/v_2/urban/ --name Sar-Opt_CGAN --direction BtoA /usr/lib/python3/dist-packages/requests/init.py:87: RequestsDependencyWarning: urllib3 (2.2.1) or chardet (4.0.0) doesn't match a supported version! warnings.warn("urllib3 ({}) or chardet ({}) doesn't match a supported " single single data.single_dataset ----------------- Options --------------- aspect_ratio: 1.0
batch_size: 1
checkpoints_dir: ./checkpoints
crop_size: 256
dataroot: ./Custom_data/v_2/urban/ [default: None] dataset_mode: single
direction: BtoA [default: AtoB] display_winsize: 256
epoch: latest
eval: False
gpu_ids: 0
init_gain: 0.02
init_type: normal
input_nc: 3
isTrain: False [default: None] load_iter: 0 [default: 0] load_size: 256
max_dataset_size: inf
model: test
model_suffix:
n_layers_D: 3
name: Sar-Opt_CGAN [default: experiment_name] ndf: 64
netD: basic
netG: resnet_9blocks
ngf: 64
no_dropout: False
no_flip: False
norm: instance
num_test: 50
num_threads: 4
output_nc: 3
phase: test
preprocess: resize_and_crop
results_dir: ./results/
serial_batches: False
suffix:
use_wandb: False
verbose: False
wandb_project_name: CycleGAN-and-pix2pix
----------------- End ------------------- single data.single_dataset dataset [SingleDataset] was created 9511 initialize network with normal model [TestModel] was created loading the model from ./checkpoints/Sar-Opt_CGAN/latest_net_G.pth Traceback (most recent call last): File "/home/Desktop/Anuj/Projects/Cyclegan&_pix2pix/pytorch-CycleGAN-and-pix2pix/test.py", line 52, in model.setup(opt) # regular setup: load and print networks; create schedulers File "/home/Desktop/Anuj/Projects/Cyclegan&_pix2pix/pytorch-CycleGAN-and-pix2pix/models/base_model.py", line 88, in setup self.load_networks(load_suffix) File "/home/Desktop/Anuj/Projects/Cyclegan&_pix2pix/pytorch-CycleGAN-and-pix2pix/models/base_model.py", line 199, in load_networks net.load_state_dict(state_dict) File "/home/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 2153, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for ResnetGenerator: Missing key(s) in state_dict: "model.10.conv_block.6.weight", "model.10.conv_block.6.bias", "model.11.conv_block.6.weight", "model.11.conv_block.6.bias", "model.12.conv_block.6.weight", "model.12.conv_block.6.bias", "model.13.conv_block.6.weight", "model.13.conv_block.6.bias", "model.14.conv_block.6.weight", "model.14.conv_block.6.bias", "model.15.conv_block.6.weight", "model.15.conv_block.6.bias", "model.16.conv_block.6.weight", "model.16.conv_block.6.bias", "model.17.conv_block.6.weight", "model.17.conv_block.6.bias", "model.18.conv_block.6.weight", "model.18.conv_block.6.bias". Unexpected key(s) in state_dict: "model.10.conv_block.5.weight", "model.10.conv_block.5.bias", "model.11.conv_block.5.weight", "model.11.conv_block.5.bias", "model.12.conv_block.5.weight", "model.12.conv_block.5.bias", "model.13.conv_block.5.weight", "model.13.conv_block.5.bias", "model.14.conv_block.5.weight", "model.14.conv_block.5.bias", "model.15.conv_block.5.weight", "model.15.conv_block.5.bias", "model.16.conv_block.5.weight", "model.16.conv_block.5.bias", "model.17.conv_block.5.weight", "model.17.conv_block.5.bias", "model.18.conv_block.5.weight", "model.18.conv_block.5.bias".

I have also changes the latest_netG*.pth -> latest_net_G.pth

please assist me with this issue.

[kanekiCY]

@arrrrr3186 add --no_dropout can solve the problem.