Views
No views yet
1import torch 2from unet import UNet # Replace with actual model import 3model = UNet() 4model.load_state_dict(torch.load("unet.pth")) 5model.eval()