File tree Expand file tree Collapse file tree 8 files changed +17
-37
lines changed
template-text-classification
template-vision-classification
template-vision-segmentation Expand file tree Collapse file tree 8 files changed +17
-37
lines changed Original file line number Diff line number Diff line change 1
1
dist
2
2
pnpm-lock.yaml
3
+ ** /__DEV_CONFIG__.json
Original file line number Diff line number Diff line change 2
2
"name" : " @pytorch-ignite/code-generator" ,
3
3
"version" : " 0.3.0" ,
4
4
"scripts" : {
5
- "dev" : " vite" ,
5
+ "dev" : " vite --port 5000 " ,
6
6
"build" : " vite build" ,
7
7
"serve" : " vite preview" ,
8
- "test" : " jest --color --runInBand" ,
9
- "test:ci" : " rm -rf ./dist-tests && start-server-and-test --expect 200 serve http://127.0.0.1:5000 test" ,
8
+ "test" : " rm -rf ./dist-tests && jest --color --runInBand" ,
9
+ "test:ci" : " start-server-and-test --expect 200 serve http://127.0.0.1:5000 test" ,
10
10
"release" : " node scripts/release.js" ,
11
11
"fmt" : " prettier --write . && bash scripts/run_code_style.sh fmt" ,
12
12
"min_lint" : " prettier --check . && bash scripts/run_code_style.sh min_lint" ,
Original file line number Diff line number Diff line change @@ -6,7 +6,7 @@ if [ $1 == "lint" ]; then
6
6
# Check that ./dist-tests/ exists and code is unzipped
7
7
ls ./dist-tests/vision-classification-all/main.py
8
8
ufmt diff .
9
- flake8 --select F401 ./dist-tests # find unused imports
9
+ flake8 --select F401,F821 ./dist-tests # find unused imports and non imported objects
10
10
elif [ $1 == " min_lint" ]; then
11
11
ufmt diff .
12
12
elif [ $1 == " fmt" ]; then
Original file line number Diff line number Diff line change 16
16
17
17
#::: } :::#
18
18
#::: if (it.save_training || it.save_evaluation) { :::#
19
- from ignite .handlers import Checkpoint , DiskSaver , global_step_from_engine
19
+ from ignite .handlers import (
20
+ Checkpoint ,
21
+ DiskSaver ,
22
+ global_step_from_engine ,
23
+ ) # usort: skip
24
+
25
+ #::: } else { :::#
26
+ from ignite .handlers import Checkpoint
20
27
21
28
#::: } :::#
22
29
#::: if (it.patience) { :::#
Original file line number Diff line number Diff line change 2
2
3
3
import ignite .distributed as idist
4
4
import torch
5
-
6
- #::: if(it.deterministic) { :::#
7
- from ignite .engine import DeterministicEngine , Engine , Events # usort: skip
8
-
9
- #::: } else { :::#
10
- from ignite .engine import Engine , Events
11
-
12
- #::: } :::#
5
+ from ignite .engine import DeterministicEngine , Engine , Events
13
6
from ignite .metrics .metric import Metric
14
7
from torch import nn
15
8
from torch .cuda .amp import autocast , GradScaler
Original file line number Diff line number Diff line change 2
2
3
3
import ignite .distributed as idist
4
4
import torch
5
-
6
- #::: if(it.deterministic) { :::#
7
- from ignite .engine import DeterministicEngine , Engine , Events # usort: skip
8
-
9
- #::: } else { :::#
10
- from ignite .engine import Engine , Events
11
-
12
- #::: } :::#
5
+ from ignite .engine import DeterministicEngine , Engine , Events
13
6
from torch .cuda .amp import autocast
14
7
from torch .nn import Module
15
8
from torch .optim import Optimizer
Original file line number Diff line number Diff line change 2
2
3
3
import ignite .distributed as idist
4
4
import torch
5
-
6
- #::: if(it.deterministic) { :::#
7
- from ignite .engine import DeterministicEngine , Engine , Events # usort: skip
8
-
9
- #::: } else { :::#
10
- from ignite .engine import Engine , Events
11
-
12
- #::: } :::#
5
+ from ignite .engine import DeterministicEngine , Engine , Events
13
6
from torch .cuda .amp import autocast
14
7
from torch .nn import Module
15
8
from torch .optim import Optimizer
Original file line number Diff line number Diff line change 3
3
import ignite .distributed as idist
4
4
import torch
5
5
from data import prepare_image_mask
6
-
7
- #::: if(it.deterministic) { :::#
8
- from ignite .engine import DeterministicEngine , Engine , Events # usort: skip
9
-
10
- #::: } else { :::#
11
- from ignite .engine import Engine , Events
12
-
13
- #::: } :::#
6
+ from ignite .engine import DeterministicEngine , Engine , Events
14
7
from ignite .metrics import Metric
15
8
from torch .cuda .amp import autocast , GradScaler
16
9
from torch .nn import Module
You can’t perform that action at this time.
0 commit comments