When I load the model, error no volumes in dictionary

[AbdulrahmanAlsumaih]

I cannot use this function for inference

def load_params_vol(self, path, device):
ckpt = torch.load(path)
model = ckpt["model"]
self.volumes = model["volumes"].to(device)
self.mask_volmes = model["mask_volmes"].to(device)
self.features = model["features"].to(device)
self.implicit_surface.load_state_dict(model["implicit_surface"])
self.has_vol = True

it will not let me load your model from your GitHub, it says that there is no volumes in dictionary.

any idea what the issue is

[prstrive]

This 'load_params_vol' function is used to load the fine-tuned volume parameters as stated here. The model we provide is obtained by generalization training, and you can directly use 'torch.load' to load the model or use our '--resume' command to load the model.