Issue with matrix and model size [PyTorch]

Question

I am trying to make a model to predict insurance cost based on the individual. And this is the code for it.

import torch
import torch.nn as nn
import torch.nn