Hi  I am training a neural network on pythorch using the training / validation/ test configuration.
for the optimizer I am doing this:
optimizer = optim.Adam(model.parameters(), lr=args.lr)
scheduler = lr_scheduler.ReduceLROnPlateau(optimizer, verbose=True, patience=15, min_lr=1e-6)
but I was wondering if the  scheduler step should be taked in respect with the test loss or the validation loss, I mean after every epoch should I do:
scheduler.step(test_loss)
or
scheduler.step(val_loss)
Thank you in advance.