Skip to content

Commit

Permalink
Experiments with Keras Model subclassing API
Browse files Browse the repository at this point in the history
  • Loading branch information
osmr committed Sep 18, 2018
1 parent 76e73aa commit d484092
Show file tree
Hide file tree
Showing 10 changed files with 474 additions and 14 deletions.
4 changes: 2 additions & 2 deletions .travis.yml
Original file line number Diff line number Diff line change
Expand Up @@ -17,9 +17,9 @@ install:
- pip install flake8 # pytest # add another testing frameworks later
before_script:
# stop the build if there are Python syntax errors or undefined names
- flake8 . --count --select=E901,E999,F821,F822,F823 --show-source --exclude=./keras --statistics
- flake8 . --count --select=E901,E999,F821,F822,F823 --show-source --exclude=./keras_ --statistics
# exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
- flake8 . --count --exit-zero --max-complexity=32 --max-line-length=127 --ignore=F403,F405,E126,E127 --exclude=./pytorch/pytorchcv/models/others,./keras --statistics
- flake8 . --count --exit-zero --max-complexity=32 --max-line-length=127 --ignore=F403,F405,E126,E127 --exclude=./pytorch/pytorchcv/models/others,./keras_ --statistics
script:
- true # pytest --capture=sys # add other tests here
notifications:
Expand Down
6 changes: 4 additions & 2 deletions chainer_/chainercv2/models/mobilenet.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,10 +30,12 @@ class ConvBlock(Chain):
Number of output channels.
ksize : int or tuple/list of 2 int
Convolution window size.
stride : int or tuple/list of 2 int
stride : int or tuple/list of 2 int, default 1
Stride of the convolution.
pad : int or tuple/list of 2 int
pad : int or tuple/list of 2 int, default 0
Padding value for convolution layer.
groups : int, default 1
Number of groups.
"""
def __init__(self,
in_channels,
Expand Down
6 changes: 4 additions & 2 deletions gluon/gluoncv2/models/mobilenet.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,10 +26,12 @@ class ConvBlock(HybridBlock):
Number of output channels.
kernel_size : int or tuple/list of 2 int
Convolution window size.
strides : int or tuple/list of 2 int
strides : int or tuple/list of 2 int, default 1
Strides of the convolution.
padding : int or tuple/list of 2 int
padding : int or tuple/list of 2 int, default 0
Padding value for convolution layer.
groups : int, default 1
Number of groups.
bn_use_global_stats : bool
Whether global moving statistics is used instead of local batch-norm for BatchNorm layers.
"""
Expand Down
File renamed without changes.
File renamed without changes.
Loading

0 comments on commit d484092

Please sign in to comment.