From c8eca2acf8f2708407d2d1c71b83515d0c9d6799 Mon Sep 17 00:00:00 2001 From: Yuhao Wang Date: Sun, 24 Apr 2022 14:37:27 +0800 Subject: [PATCH] fix: examples training code --- examples/linear_regression/README.md | 2 +- examples/linear_regression/train.py | 18 ------------------ examples/mnist/README.md | 4 ++-- examples/mnist/train.py | 18 ------------------ examples/mnist/validate.py | 2 -- 5 files changed, 3 insertions(+), 41 deletions(-) delete mode 100644 examples/linear_regression/train.py delete mode 100644 examples/mnist/train.py diff --git a/examples/linear_regression/README.md b/examples/linear_regression/README.md index 614fcef..a086718 100644 --- a/examples/linear_regression/README.md +++ b/examples/linear_regression/README.md @@ -5,5 +5,5 @@ * CPU ```shell -python train.py -c linear_regression_cpu_cfg.py +easytrain -c linear_regression_cpu_cfg.py ``` diff --git a/examples/linear_regression/train.py b/examples/linear_regression/train.py deleted file mode 100644 index 5234d53..0000000 --- a/examples/linear_regression/train.py +++ /dev/null @@ -1,18 +0,0 @@ -import sys -sys.path.append('../..') -from argparse import ArgumentParser - -from easytorch import launch_training - - -def parse_args(): - parser = ArgumentParser(description='Welcome to EasyTorch!') - parser.add_argument('-c', '--cfg', help='training config', required=True) - parser.add_argument('--gpus', help='visible gpus', type=str) - return parser.parse_args() - - -if __name__ == "__main__": - args = parse_args() - - launch_training(args.cfg, args.gpus) diff --git a/examples/mnist/README.md b/examples/mnist/README.md index e34ff9a..a1a8056 100644 --- a/examples/mnist/README.md +++ b/examples/mnist/README.md @@ -5,13 +5,13 @@ * CPU ```shell -python train.py -c config\mnist_cpu_cfg.py +easytrain -c config\mnist_cpu_cfg.py ``` * GPU (1x) ```shell -python train.py -c config\mnist_1x_cfg.py --gpus 0 +easytrain -c config\mnist_1x_cfg.py --gpus 0 ``` ## Validate diff --git a/examples/mnist/train.py b/examples/mnist/train.py deleted file mode 100644 index 5234d53..0000000 --- a/examples/mnist/train.py +++ /dev/null @@ -1,18 +0,0 @@ -import sys -sys.path.append('../..') -from argparse import ArgumentParser - -from easytorch import launch_training - - -def parse_args(): - parser = ArgumentParser(description='Welcome to EasyTorch!') - parser.add_argument('-c', '--cfg', help='training config', required=True) - parser.add_argument('--gpus', help='visible gpus', type=str) - return parser.parse_args() - - -if __name__ == "__main__": - args = parse_args() - - launch_training(args.cfg, args.gpus) diff --git a/examples/mnist/validate.py b/examples/mnist/validate.py index da722c3..04e4ca5 100644 --- a/examples/mnist/validate.py +++ b/examples/mnist/validate.py @@ -1,5 +1,3 @@ -import sys -sys.path.append('../..') from argparse import ArgumentParser from easytorch import launch_runner, Runner