Vox-adv-cpk.pth.tar

def forward(self, x): # Define the forward pass...

import torch import torch.nn as nn

# Initialize the model and load the checkpoint weights model = VoxAdvModel() model.load_state_dict(checkpoint['state_dict']) Vox-adv-cpk.pth.tar

# Load the checkpoint file checkpoint = torch.load('Vox-adv-cpk.pth.tar') def forward(self, x): # Define the forward pass