Skip to content

Commit

Permalink
fix: examples training code (#53)
Browse files Browse the repository at this point in the history
* fix: examples training code

* fix
  • Loading branch information
cnstark authored Apr 24, 2022
1 parent 4b6b4cb commit 061ea3c
Show file tree
Hide file tree
Showing 9 changed files with 5 additions and 43 deletions.
2 changes: 1 addition & 1 deletion examples/linear_regression/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -5,5 +5,5 @@
* CPU

```shell
python train.py -c linear_regression_cpu_cfg.py
easytrain -c linear_regression_cpu_cfg.py
```
18 changes: 0 additions & 18 deletions examples/linear_regression/train.py

This file was deleted.

8 changes: 4 additions & 4 deletions examples/mnist/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -5,25 +5,25 @@
* CPU

```shell
python train.py -c config\mnist_cpu_cfg.py
easytrain -c configs/mnist_cpu_cfg.py
```

* GPU (1x)

```shell
python train.py -c config\mnist_1x_cfg.py --gpus 0
easytrain -c configs/mnist_1x_cfg.py --gpus 0
```

## Validate

* CPU

```shell
python validate.py -c config\mnist_cpu_cfg.py
python validate.py -c configs/mnist_cpu_cfg.py
```

* GPU (1x)

```shell
python validate.py -c config\mnist_1x_cfg.py --gpus 0
python validate.py -c configs/mnist_1x_cfg.py --gpus 0
```
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
18 changes: 0 additions & 18 deletions examples/mnist/train.py

This file was deleted.

2 changes: 0 additions & 2 deletions examples/mnist/validate.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,3 @@
import sys
sys.path.append('../..')
from argparse import ArgumentParser

from easytorch import launch_runner, Runner
Expand Down

0 comments on commit 061ea3c

Please sign in to comment.