def create_solver_instance(data_dict, dtype, device): model = TwoLayerNet(hidden_dim=200, dtype=dtype, device=device) ############################################################################## # TODO: Use a Solver instance to train a TwoLayerNet that achieves at least # # 50% accuracy on the validation set. # ############################################################################## solver = Solver(model, data_dict, batch_size=50, num_epochs=30, num_train_samples=3000, lr_decay=0.97) ############################################################################## # END OF YOUR CODE # ############################################################################## return solver
def create_solver_instance(data_dict, dtype, device):
model = TwoLayerNet(hidden_dim=200, dtype=dtype, device=device)
##############################################################################
# TODO: Use a Solver instance to train a TwoLayerNet that achieves at least #
# 50% accuracy on the validation set. #
##############################################################################
solver = None
# Replace "pass" statement with your code
solver = Solver(model, data_dict, device='cuda')
##############################################################################
# END OF YOUR CODE #
##############################################################################
return solver
def create_solver_instance(data_dict, dtype, device):
model = TwoLayerNet(input_dim=3 * 32 * 32, hidden_dim=500)
##############################################################################
# TODO: Use a Solver instance to train a TwoLayerNet that achieves at least #
# 50% accuracy on the validation set. #
##############################################################################
solver = None
# Replace "pass" statement with your code
data = data_dict
solver = Solver(model,
data,
num_epochs=30,
batch_size=100,
print_every=100)
##############################################################################
# END OF YOUR CODE #
##############################################################################
return solver
def create_solver_instance(data_dict, dtype, device):
model = TwoLayerNet(hidden_dim=300, dtype=dtype, device=device)
##############################################################################
# TODO: Use a Solver instance to train a TwoLayerNet that achieves at least #
# 50% accuracy on the validation set. #
##############################################################################
solver = None
# Replace "pass" statement with your code
solver = Solver(model,
data_dict,
update_rule=sgd,
optim_config={'learning_rate': 1e-1,},
lr_decay=0.95,
num_epochs=50,
batch_size=256,
print_every=1000,
device=device)
##############################################################################
# END OF YOUR CODE #
##############################################################################
return solver