betas=(0.9, 0.999),
eps=1e-08,
weight_decay=5e-5)
scheduler = ReduceLROnPlateau(optimizer,
'min',
factor=0.5,
verbose=True,
patience=6,
eps=1e-8,
threshold=1e-20)
print('Starting Training')
trainloss_to_file, testloss_to_file, trainMetric_to_file, testMetric_to_file, Parameters = trainer.Train(
model,
optimizer,
Trainloader,
Testloader,
epochs=None,
Train_mode=train_mode,
Model_name=model_name,
Dataset=dataset,
scheduler=scheduler)
print('Saving model...')
save_model(model,
trainMetric_to_file,
testMetric_to_file,
trainloss_to_file,
testloss_to_file,
Parameters,
model_name,
train_mode,
dataset,
plot=False,
