Loading Pre-trained custom model about clean-pvnet

[pengsida]

python run.py --type visualize --cfg_file configs/custom.yaml
The network is defined in configs/custom.yaml.
The network is https://github.com/zju3dv/clean-pvnet/blob/master/lib/networks/pvnet/__init__.py#L5

[pengsida]

If you want to train the pvnet on your custom data, please follow these steps: https://github.com/zju3dv/clean-pvnet#training-on-the-custom-object

[ishangupta3]

I already have the custom trained model already trained.

I would like to load it myself, I was asking which network class I should use to load the model.

[ishangupta3]

Pytorch requires you to set up model in this format: (I already have the model trained on my own custom dataset) I would like to load the model myself.

model = TheModelClass(*args, **kwargs)
model.load_state_dict(torch.load(PATH))