Vox-cpk.pth.tar

# Load the model model = torch.load('vox-cpk.pth.tar', map_location=torch.device('cpu'))

# Example input (this will vary widely) dummy_input = torch.randn(1, 3, 224, 224) # Example for a 3-channel 224x224 image vox-cpk.pth.tar

import torch import torch.nn as nn