inference example

Running the inference gives this output. Can we give a set of images as input and then it gives which cluster it belongs to. python scripts/DeepDPM_load_from_checkpoint.py Sequential() [tensor([3, 2, 2, 0, 2, 1, 1, 1, 3, 2, 0, 0, 3, 0, 1, 2, 3, 5, 0, 3, 3, 2, 1, 2, 5, 3, 2, 2, 5, 2, 1, 3, 2, 2, 4, 0, 3, 2, 4, 2, 1, 0, 0, 3, 3, 4, 1, 1, 2, 5, 2, 2, 0, 0, 4, 0, 3, 2, 5, 0, 0, 2, 1, 2, 4, 3, 2, 0, 0, 3, 5, 2, 2, 1, 5, 2, 2, 2, 3, 2, 2, 0, 0, 3, 0, 1, 0, 0, 4, 2, 3, 1, 2, 3, 2, 1, 2, 3, 0, 2, 0, 2, 0, 5, 0, 2, 1, 5, 2, 3, 2, 5, 0, 2, 4, 0, 1, 3, 0, 1, 3, 1, 0, 2, 5, 2, 0, 2]),

What should be output of inference

[hafsaEn]

Thank you @imranmu , I tried the same to do inference on mnist, but I get the following error:

Traceback (most recent call last): File "/home/hafsa/DeepDPM/DeepDPM_load_from_checkpoint.py", line 30, in <module> model = ClusterNetModel.load_from_checkpoint( File "/home/hafsa/.conda/envs/deepdpm/lib/python3.10/site-packages/pytorch_lightning/core/saving.py", line 157, in load_from_checkpoint model = cls._load_model_state(checkpoint, strict=strict, **kwargs) File "/home/hafsa/.conda/envs/deepdpm/lib/python3.10/site-packages/pytorch_lightning/core/saving.py", line 205, in _load_model_state model.load_state_dict(checkpoint['state_dict'], strict=strict) File "/home/hafsa/.conda/envs/deepdpm/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1497, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for ClusterNetModel: size mismatch for cluster_net.class_fc1.weight: copying a param with shape torch.Size([50, 10]) from checkpoint, the shape in current model is torch.Size([50, 784]). size mismatch for subclustering_net.class_fc1.weight: copying a param with shape torch.Size([50, 10]) from checkpoint, the shape in current model is torch.Size([50, 784]). have you came across the same error? or can you please share the content of the file scripts/DeepDPM_load_from_checkpoint.py?

[imranmu]

@hafsaEn python scripts/DeepDPM_load_from_checkpoint.py