logo

classifier-load-model

# load the model
loaded_net = load_neural_net('checkpoint_1.pth', 'eval')
  • Share