Classes | |
class | Reshape |
Functions | |
fit (model, train_loader, val_loader, num_epochs, batch_size, optimizer, criterion, save_best, scheduler) | |
predict (model, test_X, batch_size=100) | |
Variables | |
criterion = nn.BCELoss() | |
dict | load_model_custom_objects = {"optimizer": optimizer, "criterion": criterion, "train_func": fit, "predict_func": predict} |
m = torch.jit.script(net) | |
net | |
optimizer = torch.optim.Adam | |
PyTorch_Generate_CNN_Model.fit | ( | model, | |
train_loader, | |||
val_loader, | |||
num_epochs, | |||
batch_size, | |||
optimizer, | |||
criterion, | |||
save_best, | |||
scheduler | |||
) |
Definition at line 34 of file PyTorch_Generate_CNN_Model.py.
PyTorch_Generate_CNN_Model.predict | ( | model, | |
test_X, | |||
batch_size = 100 |
|||
) |
Definition at line 93 of file PyTorch_Generate_CNN_Model.py.
PyTorch_Generate_CNN_Model.criterion = nn.BCELoss() |
Definition at line 30 of file PyTorch_Generate_CNN_Model.py.
dict PyTorch_Generate_CNN_Model.load_model_custom_objects = {"optimizer": optimizer, "criterion": criterion, "train_func": fit, "predict_func": predict} |
Definition at line 116 of file PyTorch_Generate_CNN_Model.py.
PyTorch_Generate_CNN_Model.m = torch.jit.script(net) |
Definition at line 119 of file PyTorch_Generate_CNN_Model.py.
PyTorch_Generate_CNN_Model.net |
Definition at line 14 of file PyTorch_Generate_CNN_Model.py.
PyTorch_Generate_CNN_Model.optimizer = torch.optim.Adam |
Definition at line 31 of file PyTorch_Generate_CNN_Model.py.