Thansk for replying. Found the error. Solved it. The PyTorch model is a custom pretrained model storing everything.
But I am getting the following error.
RuntimeError: can’t convert CUDA tensor to numpy (it doesn’t support GPU arrays). Use .cpu() to move the tensor to host memory first.
  File "/tmp/pycharm_project_896/agents/pytorch2keras.py", line 152, in <module>
    converter.convert((3,224,224))
  File "/tmp/pycharm_project_896/agents/pytorch2keras.py", line 72, in convert
    self.kModel.layers[target_layer].set_weights([source_layer.weight.data.numpy().transpose(transpose_dims), source_layer.bias.data.numpy()])
RuntimeError: can't convert CUDA tensor to numpy (it doesn't support GPU arrays). Use .cpu() to move the tensor to host memory first.

Any help is appreciated. Thank You