RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.Got 646 and 648 in dimension 2 at /opt/conda/conda-bld/pytorch_1570710718161/work/aten/src/THC/generic/THCTensorMath.cu:7

I try to train with MEGA code for video object detection with my own dataset. After training 11580 iters, I got the below errors. What is the meaning of error?. I set my fixed size at 416*672. All my size images are multiple of 32.

tensor([4, 0, 4, 4, 4, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 4, 4], device='cuda:0')
tensor([4, 4, 4, 4, 4, 4, 4], device='cuda:0')
tensor([[-0.0704,  0.0654, -0.0611, -0.0026],
        [-0.0701,  0.0500, -0.0606, -0.0041],
        [-0.0874,  0.0346, -0.0594, -0.0034],
        [-0.0925,  0.0536, -0.0632, -0.0067],
        [-0.1265,  0.0415, -0.0526, -0.0071],
        [-0.0778,  0.0442, -0.0606, -0.0047],
        [ 0.0030,  0.0238,  0.0022,  0.0445]], device='cuda:0',
       grad_fn=<IndexBackward>)
tensor([[ 0.6105,  2.4544, -0.5866,  0.5910],
        [ 0.3955,  0.6376, -0.3017,  0.9433],
        [ 0.5938, -0.9202, -2.4492,  0.2193],
        [ 0.1095,  1.5324, -2.0952,  0.2082],
        [-1.4429,  0.1090, -2.3657,  0.6483],
        [ 0.0000,  0.0000,  0.0000,  0.0000],
        [ 0.0000,  0.0000,  0.0000,  0.0000]], device='cuda:0')
Traceback (most recent call last):
  File "tools/train_net.py", line 225, in <module>
    main()
  File "tools/train_net.py", line 218, in main
    model = train(cfg, args.local_rank, args.distributed)
  File "tools/train_net.py", line 100, in train
    arguments
  File "/home/hungle/anaconda3/envs/mega.pytorch/mega_core/engine/trainer.py", line 96, in do_train
    loss_dict = model(images, targets)
  File "/home/hungle/anaconda3/envs/MEGA/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/hungle/anaconda3/envs/MEGA/lib/python3.7/site-packages/apex/amp/_initialize.py", line 197, in new_fwd
    **applier(kwargs, input_caster))
  File "/home/hungle/anaconda3/envs/mega.pytorch/mega_core/modeling/detector/generalized_rcnn_fgfa.py", line 98, in forward
    return self._forward_train(images["cur"], images["ref"], targets)
  File "/home/hungle/anaconda3/envs/mega.pytorch/mega_core/modeling/detector/generalized_rcnn_fgfa.py", line 110, in _forward_train
    concat_imgs = torch.cat([img.tensors, *[img_ref.tensors for img_ref in imgs_ref]], dim=0)
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 646 and 648 in dimension 2 at /opt/conda/conda-bld/pytorch_1570710718161/work/aten/src/THC/generic/THCTensorMath.cu:71
Traceback (most recent call last):
  File "/home/hungle/anaconda3/envs/MEGA/lib/python3.7/runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "/home/hungle/anaconda3/envs/MEGA/lib/python3.7/runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "/home/hungle/anaconda3/envs/MEGA/lib/python3.7/site-packages/torch/distributed/launch.py", line 253, in <module>
    main()
  File "/home/hungle/anaconda3/envs/MEGA/lib/python3.7/site-packages/torch/distributed/launch.py", line 249, in main
    cmd=cmd)

I recommend you to check the size of every input.
size means including a batch size.

1 Like

Thank you for your reply.
I try to check. Can you give me some advises?

022-02-10 17:12:16,557 mega_core.utils.model_serialization INFO: backbone.body.layer1.0.bn1.bias                                         loaded from layer1.0.bn1.bias            of shape (64,)
2022-02-10 17:12:16,557 mega_core.utils.model_serialization INFO: backbone.body.layer1.0.bn1.weight                                       loaded from layer1.0.bn1.weight          of shape (64,)
2022-02-10 17:12:16,557 mega_core.utils.model_serialization INFO: backbone.body.layer1.0.bn2.bias                                         loaded from layer1.0.bn2.bias            of shape (64,)
2022-02-10 17:12:16,557 mega_core.utils.model_serialization INFO: backbone.body.layer1.0.bn2.weight                                       loaded from layer1.0.bn2.weight          of shape (64,)
2022-02-10 17:12:16,557 mega_core.utils.model_serialization INFO: backbone.body.layer1.0.bn3.bias                                         loaded from layer1.0.bn3.bias            of shape (256,)
2022-02-10 17:12:16,557 mega_core.utils.model_serialization INFO: backbone.body.layer1.0.bn3.weight                                       loaded from layer1.0.bn3.weight          of shape (256,)
2022-02-10 17:12:16,557 mega_core.utils.model_serialization INFO: backbone.body.layer1.0.conv1.weight                                     loaded from layer1.0.conv1.weight        of shape (64, 64, 1, 1)
2022-02-10 17:12:16,558 mega_core.utils.model_serialization INFO: backbone.body.layer1.0.conv2.weight                                     loaded from layer1.0.conv2.weight        of shape (64, 64, 3, 3)
2022-02-10 17:12:16,558 mega_core.utils.model_serialization INFO: backbone.body.layer1.0.conv3.weight                                     loaded from layer1.0.conv3.weight        of shape (256, 64, 1, 1)
2022-02-10 17:12:16,558 mega_core.utils.model_serialization INFO: backbone.body.layer1.0.downsample.0.weight                              loaded from layer1.0.downsample.0.weight of shape (256, 64, 1, 1)
2022-02-10 17:12:16,558 mega_core.utils.model_serialization INFO: backbone.body.layer1.0.downsample.1.bias                                loaded from layer1.0.downsample.1.bias   of shape (256,)
2022-02-10 17:12:16,558 mega_core.utils.model_serialization INFO: backbone.body.layer1.0.downsample.1.weight                              loaded from layer1.0.downsample.1.weight of shape (256,)
2022-02-10 17:12:16,558 mega_core.utils.model_serialization INFO: backbone.body.layer1.1.bn1.bias                                         loaded from layer1.1.bn1.bias            of shape (64,)
2022-02-10 17:12:16,558 mega_core.utils.model_serialization INFO: backbone.body.layer1.1.bn1.weight                                       loaded from layer1.1.bn1.weight          of shape (64,)
2022-02-10 17:12:16,558 mega_core.utils.model_serialization INFO: backbone.body.layer1.1.bn2.bias                                         loaded from layer1.1.bn2.bias            of shape (64,)
2022-02-10 17:12:16,558 mega_core.utils.model_serialization INFO: backbone.body.layer1.1.bn2.weight                                       loaded from layer1.1.bn2.weight          of shape (64,)
2022-02-10 17:12:16,558 mega_core.utils.model_serialization INFO: backbone.body.layer1.1.bn3.bias                                         loaded from layer1.1.bn3.bias            of shape (256,)
2022-02-10 17:12:16,558 mega_core.utils.model_serialization INFO: backbone.body.layer1.1.bn3.weight                                       loaded from layer1.1.bn3.weight          of shape (256,)
2022-02-10 17:12:16,558 mega_core.utils.model_serialization INFO: backbone.body.layer1.1.conv1.weight                                     loaded from layer1.1.conv1.weight        of shape (64, 256, 1, 1)
2022-02-10 17:12:16,558 mega_core.utils.model_serialization INFO: backbone.body.layer1.1.conv2.weight                                     loaded from layer1.1.conv2.weight        of shape (64, 64, 3, 3)
2022-02-10 17:12:16,559 mega_core.utils.model_serialization INFO: backbone.body.layer1.1.conv3.weight                                     loaded from layer1.1.conv3.weight        of shape (256, 64, 1, 1)
2022-02-10 17:12:16,559 mega_core.utils.model_serialization INFO: backbone.body.layer1.2.bn1.bias                                         loaded from layer1.2.bn1.bias            of shape (64,)
2022-02-10 17:12:16,559 mega_core.utils.model_serialization INFO: backbone.body.layer1.2.bn1.weight                                       loaded from layer1.2.bn1.weight          of shape (64,)
2022-02-10 17:12:16,559 mega_core.utils.model_serialization INFO: backbone.body.layer1.2.bn2.bias                                         loaded from layer1.2.bn2.bias            of shape (64,)
2022-02-10 17:12:16,559 mega_core.utils.model_serialization INFO: backbone.body.layer1.2.bn2.weight                                       loaded from layer1.2.bn2.weight          of shape (64,)
2022-02-10 17:12:16,559 mega_core.utils.model_serialization INFO: backbone.body.layer1.2.bn3.bias                                         loaded from layer1.2.bn3.bias            of shape (256,)
2022-02-10 17:12:16,559 mega_core.utils.model_serialization INFO: backbone.body.layer1.2.bn3.weight                                       loaded from layer1.2.bn3.weight          of shape (256,)
2022-02-10 17:12:16,559 mega_core.utils.model_serialization INFO: backbone.body.layer1.2.conv1.weight                                     loaded from layer1.2.conv1.weight        of shape (64, 256, 1, 1)
2022-02-10 17:12:16,559 mega_core.utils.model_serialization INFO: backbone.body.layer1.2.conv2.weight                                     loaded from layer1.2.conv2.weight        of shape (64, 64, 3, 3)
2022-02-10 17:12:16,559 mega_core.utils.model_serialization INFO: backbone.body.layer1.2.conv3.weight                                     loaded from layer1.2.conv3.weight        of shape (256, 64, 1, 1)
2022-02-10 17:12:16,559 mega_core.utils.model_serialization INFO: backbone.body.layer2.0.bn1.bias                                         loaded from layer2.0.bn1.bias            of shape (128,)
2022-02-10 17:12:16,559 mega_core.utils.model_serialization INFO: backbone.body.layer2.0.bn1.weight                                       loaded from layer2.0.bn1.weight          of shape (128,)
2022-02-10 17:12:16,560 mega_core.utils.model_serialization INFO: backbone.body.layer2.0.bn2.bias                                         loaded from layer2.0.bn2.bias            of shape (128,)
2022-02-10 17:12:16,560 mega_core.utils.model_serialization INFO: backbone.body.layer2.0.bn2.weight                                       loaded from layer2.0.bn2.weight          of shape (128,)
2022-02-10 17:12:16,560 mega_core.utils.model_serialization INFO: backbone.body.layer2.0.bn3.bias                                         loaded from layer2.0.bn3.bias            of shape (512,)
2022-02-10 17:12:16,560 mega_core.utils.model_serialization INFO: backbone.body.layer2.0.bn3.weight                                       loaded from layer2.0.bn3.weight          of shape (512,)
2022-02-10 17:12:16,560 mega_core.utils.model_serialization INFO: backbone.body.layer2.0.conv1.weight                                     loaded from layer2.0.conv1.weight        of shape (128, 256, 1, 1)
2022-02-10 17:12:16,560 mega_core.utils.model_serialization INFO: backbone.body.layer2.0.conv2.weight                                     loaded from layer2.0.conv2.weight        of shape (128, 128, 3, 3)
2022-02-10 17:12:16,560 mega_core.utils.model_serialization INFO: backbone.body.layer2.0.conv3.weight                                     loaded from layer2.0.conv3.weight        of shape (512, 128, 1, 1)
2022-02-10 17:12:16,560 mega_core.utils.model_serialization INFO: backbone.body.layer2.0.downsample.0.weight                              loaded from layer2.0.downsample.0.weight of shape (512, 256, 1, 1)
2022-02-10 17:12:16,560 mega_core.utils.model_serialization INFO: backbone.body.layer2.0.downsample.1.bias                                loaded from layer2.0.downsample.1.bias   of shape (512,)
2022-02-10 17:12:16,560 mega_core.utils.model_serialization INFO: backbone.body.layer2.0.downsample.1.weight                              loaded from layer2.0.downsample.1.weight of shape (512,)
2022-02-10 17:12:16,560 mega_core.utils.model_serialization INFO: backbone.body.layer2.1.bn1.bias                                         loaded from layer2.1.bn1.bias            of shape (128,)
2022-02-10 17:12:16,560 mega_core.utils.model_serialization INFO: backbone.body.layer2.1.bn1.weight                                       loaded from layer2.1.bn1.weight          of shape (128,)
2022-02-10 17:12:16,560 mega_core.utils.model_serialization INFO: backbone.body.layer2.1.bn2.bias                                         loaded from layer2.1.bn2.bias            of shape (128,)
2022-02-10 17:12:16,561 mega_core.utils.model_serialization INFO: backbone.body.layer2.1.bn2.weight                                       loaded from layer2.1.bn2.weight          of shape (128,)
2022-02-10 17:12:16,561 mega_core.utils.model_serialization INFO: backbone.body.layer2.1.bn3.bias                                         loaded from layer2.1.bn3.bias            of shape (512,)
2022-02-10 17:12:16,561 mega_core.utils.model_serialization INFO: backbone.body.layer2.1.bn3.weight                                       loaded from layer2.1.bn3.weight          of shape (512,)
2022-02-10 17:12:16,561 mega_core.utils.model_serialization INFO: backbone.body.layer2.1.conv1.weight                                     loaded from layer2.1.conv1.weight        of shape (128, 512, 1, 1)
2022-02-10 17:12:16,561 mega_core.utils.model_serialization INFO: backbone.body.layer2.1.conv2.weight                                     loaded from layer2.1.conv2.weight        of shape (128, 128, 3, 3)
2022-02-10 17:12:16,561 mega_core.utils.model_serialization INFO: backbone.body.layer2.1.conv3.weight                                     loaded from layer2.1.conv3.weight        of shape (512, 128, 1, 1)
2022-02-10 17:12:16,561 mega_core.utils.model_serialization INFO: backbone.body.layer2.2.bn1.bias                                         loaded from layer2.2.bn1.bias            of shape (128,)
2022-02-10 17:12:16,561 mega_core.utils.model_serialization INFO: backbone.body.layer2.2.bn1.weight                                       loaded from layer2.2.bn1.weight          of shape (128,)
2022-02-10 17:12:16,561 mega_core.utils.model_serialization INFO: backbone.body.layer2.2.bn2.bias                                         loaded from layer2.2.bn2.bias            of shape (128,)
2022-02-10 17:12:16,561 mega_core.utils.model_serialization INFO: backbone.body.layer2.2.bn2.weight                                       loaded from layer2.2.bn2.weight          of shape (128,)
2022-02-10 17:12:16,561 mega_core.utils.model_serialization INFO: backbone.body.layer2.2.bn3.bias                                         loaded from layer2.2.bn3.bias            of shape (512,)
2022-02-10 17:12:16,561 mega_core.utils.model_serialization INFO: backbone.body.layer2.2.bn3.weight                                       loaded from layer2.2.bn3.weight          of shape (512,)
2022-02-10 17:12:16,561 mega_core.utils.model_serialization INFO: backbone.body.layer2.2.conv1.weight                                     loaded from layer2.2.conv1.weight        of shape (128, 512, 1, 1)
2022-02-10 17:12:16,562 mega_core.utils.model_serialization INFO: backbone.body.layer2.2.conv2.weight                                     loaded from layer2.2.conv2.weight        of shape (128, 128, 3, 3)
2022-02-10 17:12:16,562 mega_core.utils.model_serialization INFO: backbone.body.layer2.2.conv3.weight                                     loaded from layer2.2.conv3.weight        of shape (512, 128, 1, 1)
2022-02-10 17:12:16,562 mega_core.utils.model_serialization INFO: backbone.body.layer2.3.bn1.bias                                         loaded from layer2.3.bn1.bias            of shape (128,)
2022-02-10 17:12:16,562 mega_core.utils.model_serialization INFO: backbone.body.layer2.3.bn1.weight                                       loaded from layer2.3.bn1.weight          of shape (128,)
2022-02-10 17:12:16,562 mega_core.utils.model_serialization INFO: backbone.body.layer2.3.bn2.bias                                         loaded from layer2.3.bn2.bias            of shape (128,)
2022-02-10 17:12:16,562 mega_core.utils.model_serialization INFO: backbone.body.layer2.3.bn2.weight                                       loaded from layer2.3.bn2.weight          of shape (128,)
2022-02-10 17:12:16,562 mega_core.utils.model_serialization INFO: backbone.body.layer2.3.bn3.bias                                         loaded from layer2.3.bn3.bias            of shape (512,)
2022-02-10 17:12:16,562 mega_core.utils.model_serialization INFO: backbone.body.layer2.3.bn3.weight                                       loaded from layer2.3.bn3.weight          of shape (512,)
2022-02-10 17:12:16,562 mega_core.utils.model_serialization INFO: backbone.body.layer2.3.conv1.weight                                     loaded from layer2.3.conv1.weight        of shape (128, 512, 1, 1)
2022-02-10 17:12:16,562 mega_core.utils.model_serialization INFO: backbone.body.layer2.3.conv2.weight                                     loaded from layer2.3.conv2.weight        of shape (128, 128, 3, 3)
2022-02-10 17:12:16,562 mega_core.utils.model_serialization INFO: backbone.body.layer2.3.conv3.weight                                     loaded from layer2.3.conv3.weight        of shape (512, 128, 1, 1)
2022-02-10 17:12:16,562 mega_core.utils.model_serialization INFO: backbone.body.layer3.0.bn1.bias                                         loaded from layer3.0.bn1.bias            of shape (256,)
2022-02-10 17:12:16,562 mega_core.utils.model_serialization INFO: backbone.body.layer3.0.bn1.weight                                       loaded from layer3.0.bn1.weight          of shape (256,)
2022-02-10 17:12:16,563 mega_core.utils.model_serialization INFO: backbone.body.layer3.0.bn2.bias                                         loaded from layer3.0.bn2.bias            of shape (256,)
2022-02-10 17:12:16,563 mega_core.utils.model_serialization INFO: backbone.body.layer3.0.bn2.weight                                       loaded from layer3.0.bn2.weight          of shape (256,)
2022-02-10 17:12:16,563 mega_core.utils.model_serialization INFO: backbone.body.layer3.0.bn3.bias                                         loaded from layer3.0.bn3.bias            of shape (1024,)
2022-02-10 17:12:16,563 mega_core.utils.model_serialization INFO: backbone.body.layer3.0.bn3.weight                                       loaded from layer3.0.bn3.weight          of shape (1024,)
2022-02-10 17:12:16,563 mega_core.utils.model_serialization INFO: backbone.body.layer3.0.conv1.weight                                     loaded from layer3.0.conv1.weight        of shape (256, 512, 1, 1)


2022-02-10 17:12:16,572 mega_core.utils.model_serialization INFO: backbone.body.layer3.18.bn2.weight                                      loaded from layer3.18.bn2.weight         of shape (256,)
2022-02-10 17:12:16,572 mega_core.utils.model_serialization INFO: backbone.body.layer3.18.bn3.bias                                        loaded from layer3.18.bn3.bias           of shape (1024,)
2022-02-10 17:12:16,572 mega_core.utils.model_serialization INFO: backbone.body.layer3.18.bn3.weight                                      loaded from layer3.18.bn3.weight         of shape (1024,)
2022-02-10 17:12:16,572 mega_core.utils.model_serialization INFO: backbone.body.layer3.18.conv1.weight                                    loaded from layer3.18.conv1.weight       of shape (256, 1024, 1, 1)
2022-02-10 17:12:16,573 mega_core.utils.model_serialization INFO: backbone.body.layer3.18.conv2.weight                                    loaded from layer3.18.conv2.weight       of shape (256, 256, 3, 3)
2022-02-10 17:12:16,573 mega_core.utils.model_serialization INFO: backbone.body.layer3.18.conv3.weight                                    loaded from layer3.18.conv3.weight       of shape (1024, 256, 1, 1)
2022-02-10 17:12:16,573 mega_core.utils.model_serialization INFO: backbone.body.layer3.19.bn1.bias                                        loaded from layer3.19.bn1.bias           of shape (256,)
2022-02-10 17:12:16,573 mega_core.utils.model_serialization INFO: backbone.body.layer3.19.bn1.weight                                      loaded from layer3.19.bn1.weight         of shape (256,)
2022-02-10 17:12:16,573 mega_core.utils.model_serialization INFO: backbone.body.layer3.19.bn2.bias                                        loaded from layer3.19.bn2.bias           of shape (256,)
2022-02-10 17:12:16,573 mega_core.utils.model_serialization INFO: backbone.body.layer3.19.bn2.weight                                      loaded from layer3.19.bn2.weight         of shape (256,)
2022-02-10 17:12:16,573 mega_core.utils.model_serialization INFO: backbone.body.layer3.19.bn3.bias                                        loaded from layer3.19.bn3.bias           of shape (1024,)
2022-02-10 17:12:16,573 mega_core.utils.model_serialization INFO: backbone.body.layer3.19.bn3.weight                                      loaded from layer3.19.bn3.weight         of shape (1024,)
2022-02-10 17:12:16,573 mega_core.utils.model_serialization INFO: backbone.body.layer3.19.conv1.weight                                    loaded from layer3.19.conv1.weight       of shape (256, 1024, 1, 1)
2022-02-10 17:12:16,573 mega_core.utils.model_serialization INFO: backbone.body.layer3.19.conv2.weight                                    loaded from layer3.19.conv2.weight       of shape (256, 256, 3, 3)
2022-02-10 17:12:16,573 mega_core.utils.model_serialization INFO: backbone.body.layer3.19.conv3.weight                                    loaded from layer3.19.conv3.weight       of shape (1024, 256, 1, 1)
2022-02-10 17:12:16,573 mega_core.utils.model_serialization INFO: backbone.body.layer3.2.bn1.bias                                         loaded from layer3.2.bn1.bias            of shape (256,)
2022-02-10 17:12:16,573 mega_core.utils.model_serialization INFO: backbone.body.layer3.2.bn1.weight                                       loaded from layer3.2.bn1.weight          of shape (256,)
2022-02-10 17:12:16,574 mega_core.utils.model_serialization INFO: backbone.body.layer3.2.bn2.bias                                         loaded from layer3.2.bn2.bias            of shape (256,)
2022-02-10 17:12:16,574 mega_core.utils.model_serialization INFO: backbone.body.layer3.2.bn2.weight                                       loaded from layer3.2.bn2.weight          of shape (256,)
2022-02-10 17:12:16,574 mega_core.utils.model_serialization INFO: backbone.body.layer3.2.bn3.bias                                         loaded from layer3.2.bn3.bias            of shape (1024,)

2022-02-10 17:12:16,574 mega_core.utils.model_serialization INFO: backbone.body.layer3.2.bn3.weight                                       loaded from layer3.2.bn3.weight          of shape (1024,)
2022-02-10 17:12:16,574 mega_core.utils.model_serialization INFO: backbone.body.layer3.2.conv1.weight                                     loaded from layer3.2.conv1.weight        of shape (256, 1024, 1, 1)
2022-02-10 17:12:16,574 mega_core.utils.model_serialization INFO: backbone.body.layer3.2.conv2.weight                                     loaded from layer3.2.conv2.weight        of shape (256, 256, 3, 3)
2022-02-10 17:12:16,574 mega_core.utils.model_serialization INFO: backbone.body.layer3.2.conv3.weight                                     loaded from layer3.2.conv3.weight        of shape (1024, 256, 1, 1)
2022-02-10 17:12:16,574 mega_core.utils.model_serialization INFO: backbone.body.layer3.20.bn1.bias                                        loaded from layer3.20.bn1.bias           of shape (256,)
2022-02-10 17:12:16,574 mega_core.utils.model_serialization INFO: backbone.body.layer3.20.bn1.weight                                      loaded from layer3.20.bn1.weight         of shape (256,)
2022-02-10 17:12:16,574 mega_core.utils.model_serialization INFO: backbone.body.layer3.20.bn2.bias                                        loaded from layer3.20.bn2.bias           of shape (256,)
2022-02-10 17:12:16,574 mega_core.utils.model_serialization INFO: backbone.body.layer3.20.bn2.weight                                      loaded from layer3.20.bn2.weight         of shape (256,)
2022-02-10 17:12:16,574 mega_core.utils.model_serialization INFO: backbone.body.layer3.20.bn3.bias                                        loaded from layer3.20.bn3.bias           of shape (1024,)
2022-02-10 17:12:16,574 mega_core.utils.model_serialization INFO: backbone.body.layer3.20.bn3.weight                                      loaded from layer3.20.bn3.weight         of shape (1024,)
2022-02-10 17:12:16,575 mega_core.utils.model_serialization INFO: backbone.body.layer3.20.conv1.weight                                    loaded from layer3.20.conv1.weight       of shape (256, 1024, 1, 1)
2022-02-10 17:12:16,575 mega_core.utils.model_serialization INFO: backbone.body.layer3.20.conv2.weight                                    loaded from layer3.20.conv2.weight       of shape (256, 256, 3, 3)
2022-02-10 17:12:16,575 mega_core.utils.model_serialization INFO: backbone.body.layer3.20.conv3.weight                                    loaded from layer3.20.conv3.weight       of shape (1024, 256, 1, 1)
2022-02-10 17:12:16,575 mega_core.utils.model_serialization INFO: backbone.body.layer3.21.bn1.bias                                        loaded from layer3.21.bn1.bias           of shape (256,)
2022-02-10 17:12:16,575 mega_core.utils.model_serialization INFO: backbone.body.layer3.21.bn1.weight                                      loaded from layer3.21.bn1.weight         of shape (256,)
2022-02-10 17:12:16,575 mega_core.utils.model_serialization INFO: backbone.body.layer3.21.bn2.bias                                        loaded from layer3.21.bn2.bias           of shape (256,)
2022-02-10 17:12:16,575 mega_core.utils.model_serialization INFO: backbone.body.layer3.21.bn2.weight                                      loaded from layer3.21.bn2.weight         of shape (256,)
2022-02-10 17:12:16,575 mega_core.utils.model_serialization INFO: backbone.body.layer3.21.bn3.bias                                        loaded from layer3.21.bn3.bias           of shape (1024,)
2022-02-10 17:12:16,575 mega_core.utils.model_serialization INFO: backbone.body.layer3.21.bn3.weight                                      loaded from layer3.21.bn3.weight         of shape (1024,)
2022-02-10 17:12:16,575 mega_core.utils.model_serialization INFO: backbone.body.layer3.21.conv1.weight                                    loaded from layer3.21.conv1.weight       of shape (256, 1024, 1, 1)
2022-02-10 17:12:16,575 mega_core.utils.model_serialization INFO: backbone.body.layer3.21.conv2.weight                                    loaded from layer3.21.conv2.weight       of shape (256, 256, 3, 3)
2022-02-10 17:12:16,575 mega_core.utils.model_serialization INFO: backbone.body.layer3.21.conv3.weight                                    loaded from layer3.21.conv3.weight       of shape (1024, 256, 1, 1)
2022-02-10 17:12:16,576 mega_core.utils.model_serialization INFO: backbone.body.layer3.22.bn1.bias                                        loaded from layer3.22.bn1.bias           of shape (256,)
2022-02-10 17:12:16,576 mega_core.utils.model_serialization INFO: backbone.body.layer3.22.bn1.weight                                      loaded from layer3.22.bn1.weight         of shape (256,)
2022-02-10 17:12:16,576 mega_core.utils.model_serialization INFO: backbone.body.layer3.22.bn2.bias                                        loaded from layer3.22.bn2.bias           of shape (256,)
2022-02-10 17:12:16,576 mega_core.utils.model_serialization INFO: backbone.body.layer3.22.bn2.weight                                      loaded from layer3.22.bn2.weight         of shape (256,)
2022-02-10 17:12:16,576 mega_core.utils.model_serialization INFO: backbone.body.layer3.22.bn3.bias                                        loaded from layer3.22.bn3.bias           of shape (1024,)
2022-02-10 17:12:16,576 mega_core.utils.model_serialization INFO: backbone.body.layer3.22.bn3.weight                                      loaded from layer3.22.bn3.weight         of shape (1024,)
2022-02-10 17:12:16,576 mega_core.utils.model_serialization INFO: backbone.body.layer3.22.conv1.weight                                    loaded from layer3.22.conv1.weight       of shape (256, 1024, 1, 1)
2022-02-10 17:12:16,576 mega_core.utils.model_serialization INFO: backbone.body.layer3.22.conv2.weight                                    loaded from layer3.22.conv2.weight       of shape (256, 256, 3, 3)
2022-02-10 17:12:16,576 mega_core.utils.model_serialization INFO: backbone.body.layer3.22.conv3.weight                                    loaded from layer3.22.conv3.weight       of shape (1024, 256, 1, 1)
2022-02-10 17:12:16,576 mega_core.utils.model_serialization INFO: backbone.body.layer3.3.bn1.bias                                         loaded from layer3.3.bn1.bias            of shape (256,)
2022-02-10 17:12:16,576 mega_core.utils.model_serialization INFO: backbone.body.layer3.3.bn1.weight                                       loaded from layer3.3.bn1.weight          of shape (256,)
2022-02-10 17:12:16,576 mega_core.utils.model_serialization INFO: backbone.body.layer3.3.bn2.bias                                         loaded from layer3.3.bn2.bias            of shape (256,)
2022-02-10 17:12:16,576 mega_core.utils.model_serialization INFO: backbone.body.layer3.3.bn2.weight                                       loaded from layer3.3.bn2.weight          of shape (256,)
2022-02-10 17:12:16,577 mega_core.utils.model_serialization INFO: backbone.body.layer3.3.bn3.bias                                         loaded from layer3.3.bn3.bias            of shape (1024,)
2022-02-10 17:12:16,577 mega_core.utils.model_serialization INFO: backbone.body.layer3.3.bn3.weight                                       loaded from layer3.3.bn3.weight          of shape (1024,)
2022-02-10 17:12:16,577 mega_core.utils.model_serialization INFO: backbone.body.layer3.3.conv1.weight                                     loaded from layer3.3.conv1.weight        of shape (256, 1024, 1, 1)
2022-02-10 17:12:16,577 mega_core.utils.model_serialization INFO: backbone.body.layer3.3.conv2.weight                                     loaded from layer3.3.conv2.weight        of shape (256, 256, 3, 3)
2022-02-10 17:12:16,577 mega_core.utils.model_serialization INFO: backbone.body.layer3.3.conv3.weight                                     loaded from layer3.3.conv3.weight        of shape (1024, 256, 1, 1)
2022-02-10 17:12:16,577 mega_core.utils.model_serialization INFO: backbone.body.layer3.4.bn1.bias                                         loaded from layer3.4.bn1.bias            of shape (256,)
2022-02-10 17:12:16,577 mega_core.utils.model_serialization INFO: backbone.body.layer3.4.bn1.weight                                       loaded from layer3.4.bn1.weight          of shape (256,)
2022-02-10 17:12:16,577 mega_core.utils.model_serialization INFO: backbone.body.layer3.4.bn2.bias                                         loaded from layer3.4.bn2.bias            of shape (256,)
2022-02-10 17:12:16,577 mega_core.utils.model_serialization INFO: backbone.body.layer3.4.bn2.weight                                       loaded from layer3.4.bn2.weight          of shape (256,)
2022-02-10 17:12:16,577 mega_core.utils.model_serialization INFO: backbone.body.layer3.4.bn3.bias                                         loaded from layer3.4.bn3.bias            of shape (1024,)
2022-02-10 17:12:16,577 mega_core.utils.model_serialization INFO: backbone.body.layer3.4.bn3.weight                                       loaded from layer3.4.bn3.weight          of shape (1024,)
2022-02-10 17:12:16,577 mega_core.utils.model_serialization INFO: backbone.body.layer3.4.conv1.weight                                     loaded from layer3.4.conv1.weight        of shape (256, 1024, 1, 1)
2022-02-10 17:12:16,578 mega_core.utils.model_serialization INFO: backbone.body.layer3.4.conv2.weight                                     loaded from layer3.4.conv2.weight        of shape (256, 256, 3, 3)
2022-02-10 17:12:16,578 mega_core.utils.model_serialization INFO: backbone.body.layer3.4.conv3.weight                                     loaded from layer3.4.conv3.weight        of shape (1024, 256, 1, 1)
2022-02-10 17:12:16,578 mega_core.utils.model_serialization INFO: backbone.body.layer3.5.bn1.bias                                         loaded from layer3.5.bn1.bias            of shape (256,)
2022-02-10 17:12:16,578 mega_core.utils.model_serialization INFO: backbone.body.layer3.5.bn1.weight                                       loaded from layer3.5.bn1.weight          of shape (256,)
2022-02-10 17:12:16,578 mega_core.utils.model_serialization INFO: backbone.body.layer3.5.bn2.bias                                         loaded from layer3.5.bn2.bias            of shape (256,)
2022-02-10 17:12:16,578 mega_core.utils.model_serialization INFO: backbone.body.layer3.5.bn2.weight                                       loaded from layer3.5.bn2.weight          of shape (256,)
2022-02-10 17:12:16,578 mega_core.utils.model_serialization INFO: backbone.body.layer3.5.bn3.bias                                         loaded from layer3.5.bn3.bias            of shape (1024,)
2022-02-10 17:12:16,578 mega_core.utils.model_serialization INFO: backbone.body.layer3.5.bn3.weight                                       loaded from layer3.5.bn3.weight          of shape (1024,)
2022-02-10 17:12:16,578 mega_core.utils.model_serialization INFO: backbone.body.layer3.5.conv1.weight                                     loaded from layer3.5.conv1.weight        of shape (256, 1024, 1, 1)
2022-02-10 17:12:16,578 mega_core.utils.model_serialization INFO: backbone.body.layer3.5.conv2.weight                                     loaded from layer3.5.conv2.weight        of shape (256, 256, 3, 3)
2022-02-10 17:12:16,578 mega_core.utils.model_serialization INFO: backbone.body.layer3.5.conv3.weight                                     loaded from layer3.5.conv3.weight        of shape (1024, 256, 1, 1)
2022-02-10 17:12:16,578 mega_core.utils.model_serialization INFO: backbone.body.layer3.6.bn1.bias                                         loaded from layer3.6.bn1.bias            of shape (256,)
2022-02-10 17:12:16,579 mega_core.utils.model_serialization INFO: backbone.body.layer3.6.bn1.weight                                       loaded from layer3.6.bn1.weight          of shape (256,)
2022-02-10 17:12:16,579 mega_core.utils.model_serialization INFO: backbone.body.layer3.6.bn2.bias                                         loaded from layer3.6.bn2.bias            of shape (256,)
2022-02-10 17:12:16,579 mega_core.utils.model_serialization INFO: backbone.body.layer3.6.bn2.weight                                       loaded from layer3.6.bn2.weight          of shape (256,)
2022-02-10 17:12:16,579 mega_core.utils.model_serialization INFO: backbone.body.layer3.6.bn3.bias                                         loaded from layer3.6.bn3.bias            of shape (1024,)
2022-02-10 17:12:16,579 mega_core.utils.model_serialization INFO: backbone.body.layer3.6.bn3.weight                                       loaded from layer3.6.bn3.weight          of shape (1024,)
2022-02-10 17:12:16,579 mega_core.utils.model_serialization INFO: backbone.body.layer3.6.conv1.weight                                     loaded from layer3.6.conv1.weight        of shape (256, 1024, 1, 1)
2022-02-10 17:12:16,579 mega_core.utils.model_serialization INFO: backbone.body.layer3.6.conv2.weight                                     loaded from layer3.6.conv2.weight        of shape (256, 256, 3, 3)
2022-02-10 17:12:16,579 mega_core.utils.model_serialization INFO: backbone.body.layer3.6.conv3.weight                                     loaded from layer3.6.conv3.weight        of shape (1024, 256, 1, 1)
2022-02-10 17:12:16,579 mega_core.utils.model_serialization INFO: backbone.body.layer3.7.bn1.bias                                         loaded from layer3.7.bn1.bias            of shape (256,)
2022-02-10 17:12:16,579 mega_core.utils.model_serialization INFO: backbone.body.layer3.7.bn1.weight                                       loaded from layer3.7.bn1.weight          of shape (256,)
2022-02-10 17:12:16,579 mega_core.utils.model_serialization INFO: backbone.body.layer3.7.bn2.bias                                         loaded from layer3.7.bn2.bias            of shape (256,)
2022-02-10 17:12:16,579 mega_core.utils.model_serialization INFO: backbone.body.layer3.7.bn2.weight                                       loaded from layer3.7.bn2.weight          of shape (256,)
2022-02-10 17:12:16,579 mega_core.utils.model_serialization INFO: backbone.body.layer3.7.bn3.bias                                         loaded from layer3.7.bn3.bias            of shape (1024,)
2022-02-10 17:12:16,580 mega_core.utils.model_serialization INFO: backbone.body.layer3.7.bn3.weight                                       loaded from layer3.7.bn3.weight          of shape (1024,)
2022-02-10 17:12:16,580 mega_core.utils.model_serialization INFO: backbone.body.layer3.7.conv1.weight                                     loaded from layer3.7.conv1.weight        of shape (256, 1024, 1, 1)
2022-02-10 17:12:16,580 mega_core.utils.model_serialization INFO: backbone.body.layer3.7.conv2.weight                                     loaded from layer3.7.conv2.weight        of shape (256, 256, 3, 3)
2022-02-10 17:12:16,580 mega_core.utils.model_serialization INFO: backbone.body.layer3.7.conv3.weight                                     loaded from layer3.7.conv3.weight        of shape (1024, 256, 1, 1)
2022-02-10 17:12:16,580 mega_core.utils.model_serialization INFO: backbone.body.layer3.8.bn1.bias                                         loaded from layer3.8.bn1.bias            of shape (256,)
2022-02-10 17:12:16,580 mega_core.utils.model_serialization INFO: backbone.body.layer3.8.bn1.weight                                       loaded from layer3.8.bn1.weight          of shape (256,)
2022-02-10 17:12:16,580 mega_core.utils.model_serialization INFO: backbone.body.layer3.8.bn2.bias                                         loaded from layer3.8.bn2.bias            of shape (256,)
2022-02-10 17:12:16,580 mega_core.utils.model_serialization INFO: backbone.body.layer3.8.bn2.weight                                       loaded from layer3.8.bn2.weight          of shape (256,)
2022-02-10 17:12:16,580 mega_core.utils.model_serialization INFO: backbone.body.layer3.8.bn3.bias                                         loaded from layer3.8.bn3.bias            of shape (1024,)
2022-02-10 17:12:16,580 mega_core.utils.model_serialization INFO: backbone.body.layer3.8.bn3.weight                                       loaded from layer3.8.bn3.weight          of shape (1024,)
2022-02-10 17:12:16,580 mega_core.utils.model_serialization INFO: backbone.body.layer3.8.conv1.weight                                     loaded from layer3.8.conv1.weight        of shape (256, 1024, 1, 1)
2022-02-10 17:12:16,580 mega_core.utils.model_serialization INFO: backbone.body.layer3.8.conv2.weight                                     loaded from layer3.8.conv2.weight        of shape (256, 256, 3, 3)
2022-02-10 17:12:16,581 mega_core.utils.model_serialization INFO: backbone.body.layer3.8.conv3.weight                                     loaded from layer3.8.conv3.weight        of shape (1024, 256, 1, 1)
2022-02-10 17:12:16,581 mega_core.utils.model_serialization INFO: backbone.body.layer3.9.bn1.bias                                         loaded from layer3.9.bn1.bias            of shape (256,)
2022-02-10 17:12:16,581 mega_core.utils.model_serialization INFO: backbone.body.layer3.9.bn1.weight                                       loaded from layer3.9.bn1.weight          of shape (256,)
2022-02-10 17:12:16,581 mega_core.utils.model_serialization INFO: backbone.body.layer3.9.bn2.bias                                         loaded from layer3.9.bn2.bias            of shape (256,)
2022-02-10 17:12:16,581 mega_core.utils.model_serialization INFO: backbone.body.layer3.9.bn2.weight                                       loaded from layer3.9.bn2.weight          of shape (256,)
2022-02-10 17:12:16,581 mega_core.utils.model_serialization INFO: backbone.body.layer3.9.bn3.bias                                         loaded from layer3.9.bn3.bias            of shape (1024,)
2022-02-10 17:12:16,581 mega_core.utils.model_serialization INFO: backbone.body.layer3.9.bn3.weight                                       loaded from layer3.9.bn3.weight          of shape (1024,)
2022-02-10 17:12:16,581 mega_core.utils.model_serialization INFO: backbone.body.layer3.9.conv1.weight                                     loaded from layer3.9.conv1.weight        of shape (256, 1024, 1, 1)
2022-02-10 17:12:16,581 mega_core.utils.model_serialization INFO: backbone.body.layer3.9.conv2.weight                                     loaded from layer3.9.conv2.weight        of shape (256, 256, 3, 3)
2022-02-10 17:12:16,581 mega_core.utils.model_serialization INFO: backbone.body.layer3.9.conv3.weight                                     loaded from layer3.9.conv3.weight        of shape (1024, 256, 1, 1)
2022-02-10 17:12:16,581 mega_core.utils.model_serialization INFO: backbone.body.stem.bn1.bias                                             loaded from bn1.bias                     of shape (64,)
2022-02-10 17:12:16,581 mega_core.utils.model_serialization INFO: backbone.body.stem.bn1.weight                                           loaded from bn1.weight                   of shape (64,)
2022-02-10 17:12:16,581 mega_core.utils.model_serialization INFO: backbone.body.stem.conv1.weight                                         loaded from conv1.weight                 of shape (64, 3, 7, 7)
2022-02-10 17:12:16,582 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.0.bn1.bias                  loaded from layer4.0.bn1.bias            of shape (512,)
2022-02-10 17:12:16,582 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.0.bn1.weight                loaded from layer4.0.bn1.weight          of shape (512,)
2022-02-10 17:12:16,582 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.0.bn2.bias                  loaded from layer4.0.bn2.bias            of shape (512,)
2022-02-10 17:12:16,582 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.0.bn2.weight                loaded from layer4.0.bn2.weight          of shape (512,)
2022-02-10 17:12:16,582 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.0.bn3.bias                  loaded from layer4.0.bn3.bias            of shape (2048,)
2022-02-10 17:12:16,582 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.0.bn3.weight                loaded from layer4.0.bn3.weight          of shape (2048,)
2022-02-10 17:12:16,585 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.0.conv1.weight              loaded from layer4.0.conv1.weight        of shape (512, 1024, 1, 1)
2022-02-10 17:12:16,585 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.0.conv2.weight              loaded from layer4.0.conv2.weight        of shape (512, 512, 3, 3)
2022-02-10 17:12:16,585 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.0.conv3.weight              loaded from layer4.0.conv3.weight        of shape (2048, 512, 1, 1)
2022-02-10 17:12:16,585 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.0.downsample.0.weight       loaded from layer4.0.downsample.0.weight of shape (2048, 1024, 1, 1)
2022-02-10 17:12:16,585 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.0.downsample.1.bias         loaded from layer4.0.downsample.1.bias   of shape (2048,)
2022-02-10 17:12:16,585 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.0.downsample.1.weight       loaded from layer4.0.downsample.1.weight of shape (2048,)
2022-02-10 17:12:16,586 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.1.bn1.bias                  loaded from layer4.1.bn1.bias            of shape (512,)
2022-02-10 17:12:16,586 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.1.bn1.weight                loaded from layer4.1.bn1.weight          of shape (512,)
2022-02-10 17:12:16,586 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.1.bn2.bias                  loaded from layer4.1.bn2.bias            of shape (512,)
2022-02-10 17:12:16,586 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.1.bn2.weight                loaded from layer4.1.bn2.weight          of shape (512,)
2022-02-10 17:12:16,586 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.1.bn3.bias                  loaded from layer4.1.bn3.bias            of shape (2048,)
2022-02-10 17:12:16,586 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.1.bn3.weight                loaded from layer4.1.bn3.weight          of shape (2048,)
2022-02-10 17:12:16,586 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.1.conv1.weight              loaded from layer4.1.conv1.weight        of shape (512, 2048, 1, 1)
2022-02-10 17:12:16,586 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.1.conv2.weight              loaded from layer4.1.conv2.weight        of shape (512, 512, 3, 3)
2022-02-10 17:12:16,586 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.1.conv3.weight              loaded from layer4.1.conv3.weight        of shape (2048, 512, 1, 1)
2022-02-10 17:12:16,586 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.2.bn1.bias                  loaded from layer4.2.bn1.bias            of shape (512,)
2022-02-10 17:12:16,586 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.2.bn1.weight                loaded from layer4.2.bn1.weight          of shape (512,)
2022-02-10 17:12:16,586 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.2.bn2.bias                  loaded from layer4.2.bn2.bias            of shape (512,)
2022-02-10 17:12:16,586 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.2.bn2.weight                loaded from layer4.2.bn2.weight          of shape (512,)
2022-02-10 17:12:16,587 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.2.bn3.bias                  loaded from layer4.2.bn3.bias            of shape (2048,)
2022-02-10 17:12:16,587 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.2.bn3.weight                loaded from layer4.2.bn3.weight          of shape (2048,)
2022-02-10 17:12:16,587 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.2.conv1.weight              loaded from layer4.2.conv1.weight        of shape (512, 2048, 1, 1)
2022-02-10 17:12:16,587 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.2.conv2.weight              loaded from layer4.2.conv2.weight        of shape (512, 512, 3, 3)
2022-02-10 17:12:16,587 mega_core.utils.model_serialization INFO: roi_heads.box.feature_extractor.head.layer4.2.conv3.weight              loaded from layer4.2.conv3.weight        of shape (2048, 512, 1, 1)
2022-02-10 17:12:16,666 mega_core.utils.checkpoint INFO: Loading flownet from models/flownet.ckpt
2022-02-10 17:12:17,059 mega_core.utils.model_serialization INFO: flownet.Convolution1.bias                                               loaded from Convolution1.bias        of shape (2,)
2022-02-10 17:12:17,059 mega_core.utils.model_serialization INFO: flownet.Convolution1.weight                                             loaded from Convolution1.weight      of shape (2, 1024, 3, 3)
2022-02-10 17:12:17,059 mega_core.utils.model_serialization INFO: flownet.Convolution2.bias                                               loaded from Convolution2.bias        of shape (2,)
2022-02-10 17:12:17,059 mega_core.utils.model_serialization INFO: flownet.Convolution2.weight                                             loaded from Convolution2.weight      of shape (2, 1026, 3, 3)
2022-02-10 17:12:17,059 mega_core.utils.model_serialization INFO: flownet.Convolution3.bias                                               loaded from Convolution3.bias        of shape (2,)
2022-02-10 17:12:17,059 mega_core.utils.model_serialization INFO: flownet.Convolution3.weight                                             loaded from Convolution3.weight      of shape (2, 770, 3, 3)
2022-02-10 17:12:17,059 mega_core.utils.model_serialization INFO: flownet.Convolution4.bias                                               loaded from Convolution4.bias        of shape (2,)
2022-02-10 17:12:17,059 mega_core.utils.model_serialization INFO: flownet.Convolution4.weight                                             loaded from Convolution4.weight      of shape (2, 386, 3, 3)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.Convolution5.bias                                               loaded from Convolution5.bias        of shape (2,)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.Convolution5.weight   
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv2.bias                                                      loaded from conv2.bias               of shape (128,)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv2.weight                                                    loaded from conv2.weight             of shape (128, 64, 5, 5)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv3.bias                                                      loaded from conv3.bias               of shape (256,)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv3.weight                                                    loaded from conv3.weight             of shape (256, 128, 5, 5)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv3_1.bias                                                    loaded from conv3_1.bias             of shape (256,)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv3_1.weight                                                  loaded from conv3_1.weight           of shape (256, 256, 3, 3)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv4.bias                                                      loaded from conv4.bias               of shape (512,)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv4.weight                                                    loaded from conv4.weight             of shape (512, 256, 3, 3)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv4_1.bias                                                    loaded from conv4_1.bias             of shape (512,)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv4_1.weight                                                  loaded from conv4_1.weight           of shape (512, 512, 3, 3)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv5.bias                                                      loaded from conv5.bias               of shape (512,)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv5.weight                                                    loaded from conv5.weight             of shape (512, 512, 3, 3)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv5_1.bias                                                    loaded from conv5_1.bias             of shape (512,)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv5_1.weight                                                  loaded from conv5_1.weight           of shape (512, 512, 3, 3)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv6.bias                                                      loaded from conv6.bias               of shape (1024,)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv6.weight                                                    loaded from conv6.weight             of shape (1024, 512, 3, 3)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv6_1.bias                                                    loaded from conv6_1.bias             of shape (1024,)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.conv6_1.weight                                                  loaded from conv6_1.weight           of shape (1024, 1024, 3, 3)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.deconv2.bias                                                    loaded from deconv2.bias             of shape (64,)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.deconv2.weight                                                  loaded from deconv2.weight           of shape (386, 64, 4, 4)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.deconv3.bias                                                    loaded from deconv3.bias             of shape (128,)
2022-02-10 17:12:17,060 mega_core.utils.model_serialization INFO: flownet.deconv3.weight                                                  loaded from deconv3.weight           of shape (770, 128, 4, 4)
2022-02-10 17:12:17,061 mega_core.utils.model_serialization INFO: flownet.deconv4.bias                                                    loaded from deconv4.bias             of shape (256,)
2022-02-10 17:12:17,061 mega_core.utils.model_serialization INFO: flownet.deconv4.weight                                                  loaded from deconv4.weight           of shape (1026, 256, 4, 4)
2022-02-10 17:12:17,061 mega_core.utils.model_serialization INFO: flownet.deconv5.bias                                                    loaded from deconv5.bias             of shape (512,)
2022-02-10 17:12:17,061 mega_core.utils.model_serialization INFO: flownet.deconv5.weight                                                  loaded from deconv5.weight           of shape (1024, 512, 4, 4)
2022-02-10 17:12:17,061 mega_core.utils.model_serialization INFO: flownet.flow_conv1.bias                                                 loaded from flow_conv1.bias          of shape (64,)
2022-02-10 17:12:17,061 mega_core.utils.model_serialization INFO: flownet.flow_conv1.weight                                               loaded from flow_conv1.weight        of shape (64, 6, 7, 7)
2022-02-10 17:12:17,061 mega_core.utils.model_serialization INFO: flownet.upsample_flow3to2.bias                                          loaded from upsample_flow3to2.bias   of shape (2,)
2022-02-10 17:12:17,061 mega_core.utils.model_serialization INFO: flownet.upsample_flow3to2.weight                                        loaded from upsample_flow3to2.weight of shape (2, 2, 4, 4)
2022-02-10 17:12:17,061 mega_core.utils.model_serialization INFO: flownet.upsample_flow4to3.bias                                          loaded from upsample_flow4to3.bias   of shape (2,)
2022-02-10 17:12:17,061 mega_core.utils.model_serialization INFO: flownet.upsample_flow4to3.weight                                        loaded from upsample_flow4to3.weight of shape (2, 2, 4, 4)
2022-02-10 17:12:17,061 mega_core.utils.model_serialization INFO: flownet.upsample_flow5to4.bias                                          loaded from upsample_flow5to4.bias   of shape (2,)
2022-02-10 17:12:17,061 mega_core.utils.model_serialization INFO: flownet.upsample_flow5to4.weight                                        loaded from upsample_flow5to4.weight of shape (2, 2, 4, 4)
2022-02-10 17:12:17,061 mega_core.utils.model_serialization INFO: flownet.upsample_flow6to5.bias                                          loaded from upsample_flow6to5.bias   of shape (2,)
2022-02-10 17:12:17,061 mega_core.utils.model_serialization INFO: flownet.upsample_flow6to5.weight                                        loaded from upsample_flow6to5.weight of shape (2, 2, 4, 4)

The input which I meant is images['cur'] and image['ref'].
Would you observe those tensors by .shape?