f074e591315e0fea6689d7d42c8887794bc79c28,containers/unsupported_runtime/task.py,,main,#,43

Before Change

def main():
    // Training settings
    parser = argparse.ArgumentParser(description="PyTorch Sonar Example")
    parser.add_argument("--model-dir",
                        type=str,
                        help="Where to save the model")
    parser.add_argument("--model-name",
                        type=str,
                        default="sonar_model.h5",
                        help="What to name the saved model file")
    parser.add_argument("--batch-size",
                        type=int,
                        default=4,
                        help="input batch size for training (default: 4)")
    parser.add_argument("--test-split",
                        type=float,
                        default=0.2,
                        help="split size for training / testing dataset")
    parser.add_argument("--epochs",
                        type=int,
                        default=10,
                        help="number of epochs to train (default: 10)")
    parser.add_argument("--seed",
                        type=int,
                        default=42,
                        help="random seed (default: 42)")
    args = parser.parse_args()

    task(args)

After Change

def main():
    args = get_args()
    train_model(args)


if __name__ == "__main__":

Instances