scheduler Learning rate scheduler to be used. None, CosineAnnealing, ReduceLROnPlateau, StepLR lr_warmup_steps Number of warmup steps for the learning rate scheduler. integer >= 0 weight_decay Weight decay. float train_data_path Directory path which contains the training data. string test_data...