Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
hehl2
Torchaudio
Commits
9078c0b9
"vscode:/vscode.git/clone" did not exist on "f33f9444205023cf49bf19ec960992b39663f6fd"
Unverified
Commit
9078c0b9
authored
Aug 03, 2021
by
yangarbiter
Committed by
GitHub
Aug 03, 2021
Browse files
Add style checks on example files on CI (#1667)
parent
16f3b2f9
Changes
8
Show whitespace changes
Inline
Side-by-side
Showing
8 changed files
with
13 additions
and
6 deletions
+13
-6
.circleci/unittest/linux/scripts/run_style_checks.sh
.circleci/unittest/linux/scripts/run_style_checks.sh
+1
-1
examples/interactive_asr/__init__.py
examples/interactive_asr/__init__.py
+2
-0
examples/libtorchaudio/speech_recognition/parse_voxforge.py
examples/libtorchaudio/speech_recognition/parse_voxforge.py
+1
-1
examples/source_separation/conv_tasnet/__init__.py
examples/source_separation/conv_tasnet/__init__.py
+2
-0
examples/source_separation/conv_tasnet/train.py
examples/source_separation/conv_tasnet/train.py
+3
-3
examples/source_separation/utils/__init__.py
examples/source_separation/utils/__init__.py
+2
-0
examples/source_separation/utils/dataset/__init__.py
examples/source_separation/utils/dataset/__init__.py
+2
-0
examples/source_separation/utils/dataset/utils.py
examples/source_separation/utils/dataset/utils.py
+0
-1
No files found.
.circleci/unittest/linux/scripts/run_style_checks.sh
View file @
9078c0b9
...
@@ -29,7 +29,7 @@ set +e
...
@@ -29,7 +29,7 @@ set +e
exit_status
=
0
exit_status
=
0
printf
"
\x
1b[34mRunning flake8:
\x
1b[0m
\n
"
printf
"
\x
1b[34mRunning flake8:
\x
1b[0m
\n
"
flake8 torchaudio
test
build_tools/setup_helpers docs/source/conf.py
flake8 torchaudio
test
build_tools/setup_helpers docs/source/conf.py
examples
status
=
$?
status
=
$?
exit_status
=
"
$((
exit_status+status
))
"
exit_status
=
"
$((
exit_status+status
))
"
if
[
"
${
status
}
"
-ne
0
]
;
then
if
[
"
${
status
}
"
-ne
0
]
;
then
...
...
examples/interactive_asr/__init__.py
View file @
9078c0b9
from
.
import
utils
,
vad
from
.
import
utils
,
vad
__all__
=
[
'utils'
,
'vad'
]
examples/libtorchaudio/speech_recognition/parse_voxforge.py
View file @
9078c0b9
...
@@ -11,7 +11,7 @@ example: python parse_voxforge.py voxforge/de/Helge-20150608-aku
...
@@ -11,7 +11,7 @@ example: python parse_voxforge.py voxforge/de/Helge-20150608-aku
...
...
Dataset can be obtained from http://www.repository.voxforge1.org/downloads/de/Trunk/Audio/Main/16kHz_16bit/
Dataset can be obtained from http://www.repository.voxforge1.org/downloads/de/Trunk/Audio/Main/16kHz_16bit/
"""
"""
# noqa: E501
import
os
import
os
import
argparse
import
argparse
from
pathlib
import
Path
from
pathlib
import
Path
...
...
examples/source_separation/conv_tasnet/__init__.py
View file @
9078c0b9
...
@@ -2,3 +2,5 @@ from . import (
...
@@ -2,3 +2,5 @@ from . import (
train
,
train
,
trainer
,
trainer
,
)
)
__all__
=
[
'train'
,
'trainer'
]
examples/source_separation/conv_tasnet/train.py
View file @
9078c0b9
...
@@ -63,7 +63,7 @@ def _parse_args(args):
...
@@ -63,7 +63,7 @@ def _parse_args(args):
group
.
add_argument
(
group
.
add_argument
(
"--batch-size"
,
"--batch-size"
,
type
=
int
,
type
=
int
,
help
=
f
"Batch size. (default: 16 // world_size)"
,
help
=
"Batch size. (default: 16 // world_size)"
,
)
)
group
=
parser
.
add_argument_group
(
"Training Options"
)
group
=
parser
.
add_argument_group
(
"Training Options"
)
group
.
add_argument
(
group
.
add_argument
(
...
@@ -223,7 +223,7 @@ def train(args):
...
@@ -223,7 +223,7 @@ def train(args):
optimizer
.
load_state_dict
(
checkpoint
[
"optimizer"
])
optimizer
.
load_state_dict
(
checkpoint
[
"optimizer"
])
else
:
else
:
dist_utils
.
synchronize_params
(
dist_utils
.
synchronize_params
(
str
(
args
.
save_dir
/
f
"tmp.pt"
),
device
,
model
,
optimizer
str
(
args
.
save_dir
/
"tmp.pt"
),
device
,
model
,
optimizer
)
)
lr_scheduler
=
torch
.
optim
.
lr_scheduler
.
ReduceLROnPlateau
(
lr_scheduler
=
torch
.
optim
.
lr_scheduler
.
ReduceLROnPlateau
(
...
@@ -258,7 +258,7 @@ def train(args):
...
@@ -258,7 +258,7 @@ def train(args):
debug
=
args
.
debug
,
debug
=
args
.
debug
,
)
)
log_path
=
args
.
save_dir
/
f
"log.csv"
log_path
=
args
.
save_dir
/
"log.csv"
_write_header
(
log_path
,
args
)
_write_header
(
log_path
,
args
)
dist_utils
.
write_csv_on_master
(
dist_utils
.
write_csv_on_master
(
log_path
,
log_path
,
...
...
examples/source_separation/utils/__init__.py
View file @
9078c0b9
...
@@ -3,3 +3,5 @@ from . import (
...
@@ -3,3 +3,5 @@ from . import (
dist_utils
,
dist_utils
,
metrics
,
metrics
,
)
)
__all__
=
[
'dataset'
,
'dist_utils'
,
'metrics'
]
examples/source_separation/utils/dataset/__init__.py
View file @
9078c0b9
from
.
import
utils
,
wsj0mix
from
.
import
utils
,
wsj0mix
__all__
=
[
'utils'
,
'wsj0mix'
]
examples/source_separation/utils/dataset/utils.py
View file @
9078c0b9
...
@@ -43,7 +43,6 @@ def _fix_num_frames(sample: wsj0mix.SampleType, target_num_frames: int, random_s
...
@@ -43,7 +43,6 @@ def _fix_num_frames(sample: wsj0mix.SampleType, target_num_frames: int, random_s
return
mix
,
src
,
mask
return
mix
,
src
,
mask
def
collate_fn_wsj0mix_train
(
samples
:
List
[
wsj0mix
.
SampleType
],
sample_rate
,
duration
):
def
collate_fn_wsj0mix_train
(
samples
:
List
[
wsj0mix
.
SampleType
],
sample_rate
,
duration
):
target_num_frames
=
int
(
duration
*
sample_rate
)
target_num_frames
=
int
(
duration
*
sample_rate
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment