Skip to content

Pruning 한 모델을 가지고 테스트를 하고 싶은데 어떻게 하면 될까요? #18

@godhj93

Description

@godhj93

Pruning한 모델의 성능을 검증하고자 src/trainer.py 에 있는 run_one_epoch 메서드를 아래와 같이 수정하였습니다.

    def run_one_epoch(
        self,
        epoch: int,
        extra_log_info: List[Tuple[str, float, Callable[[float], str]]] = None,
    ) -> None:
        """Train one epoch and run testing and logging."""
        self.lr_scheduler(self.optimizer, epoch)

        # train
        #train_loss, train_stat = self.train_one_epoch()

        # test
        test_loss, test_stat = self.test_one_epoch()
        print(test_loss,test_stat)
        # save all params that showed the best acc
        # test_acc = test_stat["model_acc"]
        # if test_acc > self.best_acc:
        #     self.best_acc = test_acc
        #     filename = str(epoch) + "_" + f"{self.best_acc:.2f}".replace(".", "_")
        #     self.save_params(self.model_save_dir, filename, epoch)

        # # log
        # if not extra_log_info:
        #     extra_log_info = []
        # lr = self.optimizer.param_groups[0]["lr"]
        # log_info: List[Tuple[str, float, Callable[[float], str]]] = []
        # log_info.append(("train/lr", lr, default_format))
        # log_info.append(("train/loss", train_loss, default_format))
        # log_info += [("train/" + k, v, percent_format) for k, v in train_stat.items()]
        # log_info.append(("test/loss", test_loss, default_format))
        # log_info += [("test/" + k, v, percent_format) for k, v in test_stat.items()]
        # log_info.append(("test/best_acc", self.best_acc, percent_format))
        # self.log_one_epoch(epoch, log_info + extra_log_info)

그 후 model_compression/ 에 test.py 파일을 만들고 다음과 같이 작성하였습니다.


# -*- coding: utf-8 -*-
"""Training Runner.

- Author: Curt-Park
- Email: jwpark@jmarple.ai
"""


import argparse

from src.runners import curr_time, initialize
from src.runners.trainer import Trainer

# arguments
parser = argparse.ArgumentParser(description="Model trainer.")
parser.add_argument("--multi-gpu", action="store_true", help="Multi-GPU use")
parser.add_argument("--gpu", default=0, type=int, help="GPU id to use")
parser.add_argument(
    "--finetune", type=str, default="", help="Model path to finetune (pth.tar)"
)
parser.add_argument(
    "--resume",
    type=str,
    default="",
    help="Input log directory name to resume in save/checkpoint",
)
parser.add_argument(
    "--half", dest="half", action="store_true", help="Use half precision"
)
parser.add_argument(
    "--wlog", dest="wlog", action="store_true", help="Turns on wandb logging"
)
parser.add_argument(
    "--config",
    type=str,
    default="config/train/simplenet.py",
    help="Configuration path (.py)",
)
parser.set_defaults(half=False)
parser.set_defaults(multi_gpu=False)
parser.set_defaults(wlog=False)
args = parser.parse_args()

# initialize
config, dir_prefix, device = initialize(
    "train", args.config, args.resume, args.multi_gpu, args.gpu
)
if args.resume:
    wandb_name = args.resume
    finetune = ""
else:
    wandb_name = curr_time
    finetune = args.finetune
wandb_init_params = dict(config=config, name=wandb_name, group=args.config)

# run training
trainer = Trainer(
    config=config,
    dir_prefix=dir_prefix,
    checkpt_dir="train",
    finetune=finetune,
    wandb_log=args.wlog,
    wandb_init_params=wandb_init_params,
    device=device,
    half=args.half,
)
trainer.run(args.resume)

그 후 다음과 같이 명령어를 입력했는데 아래와 같은 에러가 나옵니다.

hj@lab:~/hdd/hdd2000/add/model_compression$ python test.py --config config/train/cifar100/densenet_small.py --finetune my_work/models/LTH/171_62_87.pth.tar 


2021-03-19 01:41:03,446 - config_validator.py:44 - INFO - [Config]	SEED: 777	AUG_TRAIN: randaugment_train_cifar100	AUG_TRAIN_PARAMS: {'n_select': 2, 'level': None}	AUG_TEST: simple_augment_test_cifar100	CUTMIX: {'beta': 1, 'prob': 0.5}	DATASET: CIFAR100	MODEL_NAME: densenet	MODEL_PARAMS: {'num_classes': 100, 'inplanes': 24, 'growthRate': 12, 'compressionRate': 2, 'block_configs': (16, 16, 16)}	CRITERION: CrossEntropy	CRITERION_PARAMS: {'num_classes': 100, 'label_smoothing': 0.1}	LR_SCHEDULER: WarmupCosineLR	LR_SCHEDULER_PARAMS: {'warmup_epochs': 10, 'start_lr': 0.001, 'min_lr': 1e-05, 'n_rewinding': 1}	BATCH_SIZE: 64	LR: 0.1	MOMENTUM: 0.9	WEIGHT_DECAY: 0.0001	EPOCHS: 300	N_WORKERS: 20
Traceback (most recent call last):
  File "test.py", line 65, in <module>
    half=args.half,
  File "/hdd_ext/hdd2000/add/model_compression/src/runners/trainer.py", line 62, in __init__
    self.load_model(finetune)
  File "/hdd_ext/hdd2000/add/model_compression/src/runners/trainer.py", line 316, in load_model
    self.model, checkpt["state_dict"], with_mask=with_mask
  File "/hdd_ext/hdd2000/add/model_compression/src/models/utils.py", line 42, in initialize_params
    model.load_state_dict(model_dict)
  File "/home/hj/anaconda3/envs/ML/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1224, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for DenseNet:
	size mismatch for stem.bn.weight: copying a param with shape torch.Size([24, 3, 3, 3]) from checkpoint, the shape in current model is torch.Size([24]).
	size mismatch for dense_blocks.0.layers.0.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 24, 1, 1]).
	size mismatch for dense_blocks.0.layers.0.conv1.bn.weight: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.0.conv1.bn.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.0.conv1.bn.running_mean: copying a param with shape torch.Size([48, 24, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.0.conv1.bn.running_var: copying a param with shape torch.Size([48, 24, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.0.conv2.conv.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.0.layers.0.conv2.bn.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.0.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.0.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.0.conv2.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.1.conv1.conv.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48, 36, 1, 1]).
	size mismatch for dense_blocks.0.layers.1.conv1.bn.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.1.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.1.conv1.bn.running_mean: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.1.conv1.bn.running_var: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.1.conv2.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.0.layers.1.conv2.bn.running_mean: copying a param with shape torch.Size([48, 36, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.1.conv2.bn.running_var: copying a param with shape torch.Size([48, 36, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.2.conv1.conv.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([48, 48, 1, 1]).
	size mismatch for dense_blocks.0.layers.2.conv1.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.2.conv2.bn.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.3.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 60, 1, 1]).
	size mismatch for dense_blocks.0.layers.3.conv1.bn.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.3.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.3.conv1.bn.running_mean: copying a param with shape torch.Size([48, 48, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.3.conv1.bn.running_var: copying a param with shape torch.Size([48, 48, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.3.conv2.conv.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.0.layers.3.conv2.bn.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.3.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.3.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.3.conv2.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.4.conv1.conv.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48, 72, 1, 1]).
	size mismatch for dense_blocks.0.layers.4.conv1.bn.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.4.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.4.conv1.bn.running_mean: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.4.conv1.bn.running_var: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.4.conv2.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.0.layers.4.conv2.bn.running_mean: copying a param with shape torch.Size([48, 60, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.4.conv2.bn.running_var: copying a param with shape torch.Size([48, 60, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.5.conv1.conv.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([48, 84, 1, 1]).
	size mismatch for dense_blocks.0.layers.5.conv1.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.5.conv2.bn.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.6.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 96, 1, 1]).
	size mismatch for dense_blocks.0.layers.6.conv1.bn.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.6.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.6.conv1.bn.running_mean: copying a param with shape torch.Size([48, 72, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.6.conv1.bn.running_var: copying a param with shape torch.Size([48, 72, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.6.conv2.conv.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.0.layers.6.conv2.bn.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.6.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.6.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.6.conv2.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.7.conv1.conv.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48, 108, 1, 1]).
	size mismatch for dense_blocks.0.layers.7.conv1.bn.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.7.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.7.conv1.bn.running_mean: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.7.conv1.bn.running_var: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.7.conv2.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.0.layers.7.conv2.bn.running_mean: copying a param with shape torch.Size([48, 84, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.7.conv2.bn.running_var: copying a param with shape torch.Size([48, 84, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.8.conv1.conv.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([48, 120, 1, 1]).
	size mismatch for dense_blocks.0.layers.8.conv1.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.8.conv2.bn.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.9.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 132, 1, 1]).
	size mismatch for dense_blocks.0.layers.9.conv1.bn.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.9.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.9.conv1.bn.running_mean: copying a param with shape torch.Size([48, 96, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.9.conv1.bn.running_var: copying a param with shape torch.Size([48, 96, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.9.conv2.conv.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.0.layers.9.conv2.bn.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.9.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.9.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.9.conv2.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.10.conv1.conv.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48, 144, 1, 1]).
	size mismatch for dense_blocks.0.layers.10.conv1.bn.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.10.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.10.conv1.bn.running_mean: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.10.conv1.bn.running_var: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.10.conv2.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.0.layers.10.conv2.bn.running_mean: copying a param with shape torch.Size([48, 108, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.10.conv2.bn.running_var: copying a param with shape torch.Size([48, 108, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.11.conv1.conv.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([48, 156, 1, 1]).
	size mismatch for dense_blocks.0.layers.11.conv1.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.11.conv2.bn.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.12.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 168, 1, 1]).
	size mismatch for dense_blocks.0.layers.12.conv1.bn.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.12.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.12.conv1.bn.running_mean: copying a param with shape torch.Size([48, 120, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.12.conv1.bn.running_var: copying a param with shape torch.Size([48, 120, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.12.conv2.conv.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.0.layers.12.conv2.bn.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.12.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.12.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.12.conv2.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.13.conv1.conv.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48, 180, 1, 1]).
	size mismatch for dense_blocks.0.layers.13.conv1.bn.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.13.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.13.conv1.bn.running_mean: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.13.conv1.bn.running_var: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.13.conv2.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.0.layers.13.conv2.bn.running_mean: copying a param with shape torch.Size([48, 132, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.13.conv2.bn.running_var: copying a param with shape torch.Size([48, 132, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.14.conv1.conv.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([48, 192, 1, 1]).
	size mismatch for dense_blocks.0.layers.14.conv1.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.14.conv2.bn.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.15.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 204, 1, 1]).
	size mismatch for dense_blocks.0.layers.15.conv1.bn.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.15.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.15.conv1.bn.running_mean: copying a param with shape torch.Size([48, 144, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.15.conv1.bn.running_var: copying a param with shape torch.Size([48, 144, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.0.layers.15.conv2.conv.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.0.layers.15.conv2.bn.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.15.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.15.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.0.layers.15.conv2.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.1.conv.conv.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([108, 216, 1, 1]).
	size mismatch for dense_blocks.1.conv.bn.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([108]).
	size mismatch for dense_blocks.1.conv.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([108]).
	size mismatch for dense_blocks.1.conv.bn.running_mean: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([108]).
	size mismatch for dense_blocks.1.conv.bn.running_var: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([108]).
	size mismatch for dense_blocks.2.layers.0.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 108, 1, 1]).
	size mismatch for dense_blocks.2.layers.0.conv1.bn.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.0.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.0.conv1.bn.running_mean: copying a param with shape torch.Size([48, 156, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.0.conv1.bn.running_var: copying a param with shape torch.Size([48, 156, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.0.conv2.conv.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.2.layers.0.conv2.bn.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.0.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.0.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.0.conv2.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.1.conv1.conv.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48, 120, 1, 1]).
	size mismatch for dense_blocks.2.layers.1.conv1.bn.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.1.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.1.conv1.bn.running_mean: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.1.conv1.bn.running_var: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.1.conv2.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.2.layers.1.conv2.bn.running_mean: copying a param with shape torch.Size([48, 168, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.1.conv2.bn.running_var: copying a param with shape torch.Size([48, 168, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.2.conv1.conv.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([48, 132, 1, 1]).
	size mismatch for dense_blocks.2.layers.2.conv1.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.2.conv2.bn.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.3.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 144, 1, 1]).
	size mismatch for dense_blocks.2.layers.3.conv1.bn.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.3.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.3.conv1.bn.running_mean: copying a param with shape torch.Size([48, 180, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.3.conv1.bn.running_var: copying a param with shape torch.Size([48, 180, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.3.conv2.conv.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.2.layers.3.conv2.bn.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.3.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.3.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.3.conv2.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.4.conv1.conv.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48, 156, 1, 1]).
	size mismatch for dense_blocks.2.layers.4.conv1.bn.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.4.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.4.conv1.bn.running_mean: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.4.conv1.bn.running_var: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.4.conv2.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.2.layers.4.conv2.bn.running_mean: copying a param with shape torch.Size([48, 192, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.4.conv2.bn.running_var: copying a param with shape torch.Size([48, 192, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.5.conv1.conv.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([48, 168, 1, 1]).
	size mismatch for dense_blocks.2.layers.5.conv1.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.5.conv2.bn.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.6.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 180, 1, 1]).
	size mismatch for dense_blocks.2.layers.6.conv1.bn.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.6.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.6.conv1.bn.running_mean: copying a param with shape torch.Size([48, 204, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.6.conv1.bn.running_var: copying a param with shape torch.Size([48, 204, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.6.conv2.conv.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.2.layers.6.conv2.bn.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.6.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.6.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.6.conv2.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.7.conv1.conv.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48, 192, 1, 1]).
	size mismatch for dense_blocks.2.layers.7.conv1.bn.weight: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.7.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.7.conv1.bn.running_mean: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.7.conv1.bn.running_var: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.7.conv2.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.2.layers.7.conv2.bn.running_mean: copying a param with shape torch.Size([108, 216, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.7.conv2.bn.running_var: copying a param with shape torch.Size([108, 216, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.8.conv1.conv.weight: copying a param with shape torch.Size([108]) from checkpoint, the shape in current model is torch.Size([48, 204, 1, 1]).
	size mismatch for dense_blocks.2.layers.8.conv1.bn.weight: copying a param with shape torch.Size([108]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.8.conv1.bn.bias: copying a param with shape torch.Size([108]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.8.conv1.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.8.conv1.bn.running_var: copying a param with shape torch.Size([108]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.8.conv2.conv.weight: copying a param with shape torch.Size([48, 108, 1, 1]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.2.layers.8.conv2.bn.weight: copying a param with shape torch.Size([48, 108, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.8.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.8.conv2.bn.running_mean: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.8.conv2.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.9.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 216, 1, 1]).
	size mismatch for dense_blocks.2.layers.9.conv1.bn.running_mean: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.9.conv1.bn.running_var: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.9.conv2.conv.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.2.layers.9.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.10.conv1.conv.weight: copying a param with shape torch.Size([48, 120, 1, 1]) from checkpoint, the shape in current model is torch.Size([48, 228, 1, 1]).
	size mismatch for dense_blocks.2.layers.10.conv1.bn.weight: copying a param with shape torch.Size([48, 120, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.10.conv2.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.2.layers.10.conv2.bn.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.10.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.10.conv2.bn.running_mean: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.10.conv2.bn.running_var: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.11.conv1.conv.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48, 240, 1, 1]).
	size mismatch for dense_blocks.2.layers.11.conv1.bn.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.11.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.11.conv1.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.11.conv1.bn.running_var: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.11.conv2.conv.weight: copying a param with shape torch.Size([48, 132, 1, 1]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.2.layers.11.conv2.bn.weight: copying a param with shape torch.Size([48, 132, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.11.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.11.conv2.bn.running_mean: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.11.conv2.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.12.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 252, 1, 1]).
	size mismatch for dense_blocks.2.layers.12.conv1.bn.running_mean: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.12.conv1.bn.running_var: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.12.conv2.conv.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.2.layers.12.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.13.conv1.conv.weight: copying a param with shape torch.Size([48, 144, 1, 1]) from checkpoint, the shape in current model is torch.Size([48, 264, 1, 1]).
	size mismatch for dense_blocks.2.layers.13.conv1.bn.weight: copying a param with shape torch.Size([48, 144, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.13.conv2.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.2.layers.13.conv2.bn.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.13.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.13.conv2.bn.running_mean: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.13.conv2.bn.running_var: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.14.conv1.conv.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48, 276, 1, 1]).
	size mismatch for dense_blocks.2.layers.14.conv1.bn.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.14.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.14.conv1.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.14.conv1.bn.running_var: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.14.conv2.conv.weight: copying a param with shape torch.Size([48, 156, 1, 1]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.2.layers.14.conv2.bn.weight: copying a param with shape torch.Size([48, 156, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.14.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.14.conv2.bn.running_mean: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.14.conv2.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.2.layers.15.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 288, 1, 1]).
	size mismatch for dense_blocks.2.layers.15.conv1.bn.running_mean: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.15.conv1.bn.running_var: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.2.layers.15.conv2.conv.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.2.layers.15.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.3.conv.conv.weight: copying a param with shape torch.Size([48, 168, 1, 1]) from checkpoint, the shape in current model is torch.Size([150, 300, 1, 1]).
	size mismatch for dense_blocks.3.conv.bn.weight: copying a param with shape torch.Size([48, 168, 1, 1]) from checkpoint, the shape in current model is torch.Size([150]).
	size mismatch for dense_blocks.3.conv.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([150]).
	size mismatch for dense_blocks.3.conv.bn.running_mean: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([150]).
	size mismatch for dense_blocks.3.conv.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([150]).
	size mismatch for dense_blocks.4.layers.0.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 150, 1, 1]).
	size mismatch for dense_blocks.4.layers.0.conv1.bn.running_mean: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.0.conv1.bn.running_var: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.0.conv2.conv.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.4.layers.0.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.1.conv1.conv.weight: copying a param with shape torch.Size([48, 180, 1, 1]) from checkpoint, the shape in current model is torch.Size([48, 162, 1, 1]).
	size mismatch for dense_blocks.4.layers.1.conv1.bn.weight: copying a param with shape torch.Size([48, 180, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.1.conv2.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.4.layers.1.conv2.bn.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.1.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.1.conv2.bn.running_mean: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.1.conv2.bn.running_var: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.2.conv1.conv.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48, 174, 1, 1]).
	size mismatch for dense_blocks.4.layers.2.conv1.bn.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.2.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.2.conv1.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.2.conv1.bn.running_var: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.2.conv2.conv.weight: copying a param with shape torch.Size([48, 192, 1, 1]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.4.layers.2.conv2.bn.weight: copying a param with shape torch.Size([48, 192, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.2.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.2.conv2.bn.running_mean: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.2.conv2.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.3.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 186, 1, 1]).
	size mismatch for dense_blocks.4.layers.3.conv1.bn.running_mean: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.3.conv1.bn.running_var: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.3.conv2.conv.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.4.layers.3.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.4.conv1.conv.weight: copying a param with shape torch.Size([48, 204, 1, 1]) from checkpoint, the shape in current model is torch.Size([48, 198, 1, 1]).
	size mismatch for dense_blocks.4.layers.4.conv1.bn.weight: copying a param with shape torch.Size([48, 204, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.4.conv2.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.4.layers.4.conv2.bn.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.4.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.4.conv2.bn.running_mean: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.4.conv2.bn.running_var: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.5.conv1.conv.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48, 210, 1, 1]).
	size mismatch for dense_blocks.4.layers.5.conv1.bn.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.5.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.5.conv1.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.5.conv1.bn.running_var: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.5.conv2.conv.weight: copying a param with shape torch.Size([48, 216, 1, 1]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.4.layers.5.conv2.bn.weight: copying a param with shape torch.Size([48, 216, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.5.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.5.conv2.bn.running_mean: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.5.conv2.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.6.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 222, 1, 1]).
	size mismatch for dense_blocks.4.layers.6.conv1.bn.running_mean: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.6.conv1.bn.running_var: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.6.conv2.conv.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.4.layers.6.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.7.conv1.conv.weight: copying a param with shape torch.Size([48, 228, 1, 1]) from checkpoint, the shape in current model is torch.Size([48, 234, 1, 1]).
	size mismatch for dense_blocks.4.layers.7.conv1.bn.weight: copying a param with shape torch.Size([48, 228, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.7.conv2.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.4.layers.7.conv2.bn.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.7.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.7.conv2.bn.running_mean: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.7.conv2.bn.running_var: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.8.conv1.conv.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48, 246, 1, 1]).
	size mismatch for dense_blocks.4.layers.8.conv1.bn.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.8.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.8.conv1.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.8.conv1.bn.running_var: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.8.conv2.conv.weight: copying a param with shape torch.Size([48, 240, 1, 1]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.4.layers.8.conv2.bn.weight: copying a param with shape torch.Size([48, 240, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.8.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.8.conv2.bn.running_mean: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.8.conv2.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.9.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 258, 1, 1]).
	size mismatch for dense_blocks.4.layers.9.conv1.bn.running_mean: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.9.conv1.bn.running_var: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.9.conv2.conv.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.4.layers.9.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.10.conv1.conv.weight: copying a param with shape torch.Size([48, 252, 1, 1]) from checkpoint, the shape in current model is torch.Size([48, 270, 1, 1]).
	size mismatch for dense_blocks.4.layers.10.conv1.bn.weight: copying a param with shape torch.Size([48, 252, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.10.conv2.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.4.layers.10.conv2.bn.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.10.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.10.conv2.bn.running_mean: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.10.conv2.bn.running_var: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.11.conv1.conv.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48, 282, 1, 1]).
	size mismatch for dense_blocks.4.layers.11.conv1.bn.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.11.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.11.conv1.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.11.conv1.bn.running_var: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.11.conv2.conv.weight: copying a param with shape torch.Size([48, 264, 1, 1]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.4.layers.11.conv2.bn.weight: copying a param with shape torch.Size([48, 264, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.11.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.11.conv2.bn.running_mean: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.11.conv2.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.12.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 294, 1, 1]).
	size mismatch for dense_blocks.4.layers.12.conv1.bn.running_mean: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.12.conv1.bn.running_var: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.12.conv2.conv.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.4.layers.12.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.13.conv1.conv.weight: copying a param with shape torch.Size([48, 276, 1, 1]) from checkpoint, the shape in current model is torch.Size([48, 306, 1, 1]).
	size mismatch for dense_blocks.4.layers.13.conv1.bn.weight: copying a param with shape torch.Size([48, 276, 1, 1]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.13.conv2.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.4.layers.13.conv2.bn.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.13.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.13.conv2.bn.running_mean: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.13.conv2.bn.running_var: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.14.conv1.conv.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48, 318, 1, 1]).
	size mismatch for dense_blocks.4.layers.14.conv1.bn.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.14.conv1.bn.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.14.conv1.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.14.conv1.bn.running_var: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.14.conv2.conv.weight: copying a param with shape torch.Size([48, 288, 1, 1]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.4.layers.14.conv2.bn.weight: copying a param with shape torch.Size([48, 288, 1, 1]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.14.conv2.bn.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.14.conv2.bn.running_mean: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.14.conv2.bn.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for dense_blocks.4.layers.15.conv1.conv.weight: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([48, 330, 1, 1]).
	size mismatch for dense_blocks.4.layers.15.conv1.bn.running_mean: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.15.conv1.bn.running_var: copying a param with shape torch.Size([12, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([48]).
	size mismatch for dense_blocks.4.layers.15.conv2.conv.weight: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([12, 48, 3, 3]).
	size mismatch for dense_blocks.4.layers.15.conv2.bn.running_mean: copying a param with shape torch.Size([]) from checkpoint, the shape in current model is torch.Size([12]).
	size mismatch for fc.weight: copying a param with shape torch.Size([150, 300, 1, 1]) from checkpoint, the shape in current model is torch.Size([100, 342]).
	size mismatch for fc.bias: copying a param with shape torch.Size([150, 300, 1, 1]) from checkpoint, the shape in current model is torch.Size([100]).

모델을 추론시간과 정확도를 테스트 할 수 있는 다른 방법이 존재하나요?
존재한다면 알려주시기 부탁드립니다.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions