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