optimizer = torch.optim.Adam(model.parameters(), lr=1, weight_decay=0.0001) # scheduler = ReduceLROnPlateau(optimizer, mode='min', factor=0.8,

optimizer = torch.optim.Adam(model.parameters(), lr=1, weight_decay=0.0001) # scheduler = ReduceLROnPlateau(optimizer, mode='min', factor=0.8,