Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
PyTorch_Generate_CNN_Model.py File Reference

Classes

class  PyTorch_Generate_CNN_Model.Reshape
 

Namespaces

namespace  PyTorch_Generate_CNN_Model
 

Functions

 PyTorch_Generate_CNN_Model.fit (model, train_loader, val_loader, num_epochs, batch_size, optimizer, criterion, save_best, scheduler)
 
 PyTorch_Generate_CNN_Model.predict (model, test_X, batch_size=100)
 

Variables

 PyTorch_Generate_CNN_Model.criterion = nn.BCELoss()
 
dict PyTorch_Generate_CNN_Model.load_model_custom_objects = {"optimizer": optimizer, "criterion": criterion, "train_func": fit, "predict_func": predict}
 
 PyTorch_Generate_CNN_Model.m = torch.jit.script(net)
 
 PyTorch_Generate_CNN_Model.net
 
 PyTorch_Generate_CNN_Model.optimizer = torch.optim.Adam