TrainModel Member List

This is the complete list of members for TrainModel, including all inherited members.

__init__(self, torch.nn.Module model, torch.utils.data.DataLoader train_loader, torch.utils.data.DataLoader val_loader, torch.optim.Optimizer optimizer, torch.nn.Module criterion, int num_epochs, torch.device device, str model_dir)	TrainModel
criterion	TrainModel
device	TrainModel
load_saved_model(self, str model_file)	TrainModel
load_saved_model_dict(self, str model_file)	TrainModel
load_saved_optimizer(self, str optimizer_path)	TrainModel
model	TrainModel
model_dir	TrainModel
num_epochs	TrainModel
optimizer	TrainModel
start_time	TrainModel
test(self, torch.utils.data.DataLoader test_loader)	TrainModel
train(self)	TrainModel
train_epoch(self)	TrainModel
train_loader	TrainModel
val_loader	TrainModel
validate_epoch(self, torch.utils.data.DataLoader data_loader)	TrainModel