Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
S
seminar-breakout
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Analytics
Analytics
CI / CD
Repository
Value Stream
Wiki
Wiki
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Shashank Suhas
seminar-breakout
Commits
51239e9c
Commit
51239e9c
authored
Feb 05, 2017
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix versioning. add scripts to install.
parent
e4b0bd9d
Changes
5
Hide whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
31 additions
and
9 deletions
+31
-9
README.md
README.md
+2
-4
setup.cfg
setup.cfg
+1
-2
setup.py
setup.py
+26
-2
tensorpack/__init__.py
tensorpack/__init__.py
+1
-1
tox.ini
tox.ini
+1
-0
No files found.
README.md
View file @
51239e9c
...
@@ -38,13 +38,13 @@ Describe your training task with three components:
...
@@ -38,13 +38,13 @@ Describe your training task with three components:
2.
__DataFlow__. tensorpack allows and encourages complex data processing.
2.
__DataFlow__. tensorpack allows and encourages complex data processing.
+ All data producer has an unified
`generator`
interface, allowing them to be composed to perform complex preprocessing.
+ All data producer has an unified interface, allowing them to be composed to perform complex preprocessing.
+ Use Python to easily handle any data format, yet still keep good performance thanks to multiprocess prefetch & TF Queue prefetch.
+ Use Python to easily handle any data format, yet still keep good performance thanks to multiprocess prefetch & TF Queue prefetch.
For example, InceptionV3 can run in the same speed as the official code which reads data by TF operators.
For example, InceptionV3 can run in the same speed as the official code which reads data by TF operators.
3.
__Callbacks__, including everything you want to do apart from the training iterations, such as:
3.
__Callbacks__, including everything you want to do apart from the training iterations, such as:
+
Change hyperparameters during training
+
Change hyperparameters during training
+
Print some
variable
s of interest
+
Print some
tensor
s of interest
+
Run inference on a test dataset
+
Run inference on a test dataset
+
Run some operations once a while
+
Run some operations once a while
+
Send loss to your phone
+
Send loss to your phone
...
@@ -53,8 +53,6 @@ With the above components defined, tensorpack trainer will run the training iter
...
@@ -53,8 +53,6 @@ With the above components defined, tensorpack trainer will run the training iter
Multi-GPU training is off-the-shelf by simply switching the trainer.
Multi-GPU training is off-the-shelf by simply switching the trainer.
You can also define your own trainer for non-standard training (e.g. GAN).
You can also define your own trainer for non-standard training (e.g. GAN).
The components are designed to be independent. You can use Model or DataFlow in other projects as well.
## Dependencies:
## Dependencies:
+
Python 2 or 3
+
Python 2 or 3
...
...
setup.cfg
View file @
51239e9c
[metadata]
[metadata]
name = tensorpack
name = tensorpack
version = attr: tensorpack.__version__
author = TensorPack contributors
author = TensorPack contributors
author-email = ppwwyyxxc@gmail.com
author-email = ppwwyyxxc@gmail.com
description = Neural Network Toolbox on TensorFlow
description = Neural Network Toolbox on TensorFlow
...
@@ -11,7 +10,7 @@ license = Apache
...
@@ -11,7 +10,7 @@ license = Apache
[options]
[options]
zip_safe = False # dataset and __init__ use file
zip_safe = False # dataset and __init__ use file
packages = find:
packages = find:
# will call find_packages()
[wheel]
[wheel]
universal = 1
universal = 1
setup.py
View file @
51239e9c
from
setuptools
import
setup
from
setuptools
import
setup
import
os
import
shutil
# setup metainfo
CURRENT_DIR
=
os
.
path
.
dirname
(
__file__
)
libinfo_py
=
os
.
path
.
join
(
CURRENT_DIR
,
'tensorpack/libinfo.py'
)
libinfo
=
{
'__file__'
:
libinfo_py
}
exec
(
compile
(
open
(
libinfo_py
,
"rb"
)
.
read
(),
libinfo_py
,
'exec'
),
libinfo
,
libinfo
)
__version__
=
libinfo
[
'__version__'
]
# configure requirements
req
=
[
req
=
[
'numpy'
,
'numpy'
,
'six'
,
'six'
,
...
@@ -21,11 +31,25 @@ extra_req = [
...
@@ -21,11 +31,25 @@ extra_req = [
'tornado;python_version<"3.0"'
,
'tornado;python_version<"3.0"'
,
]
]
# TODO:
# parse scripts
# setup_requires, scripts
scripts
=
[
'scripts/plot-point.py'
,
'scripts/dump-model-params.py'
]
scripts_to_install
=
[]
for
s
in
scripts
:
dirname
=
os
.
path
.
dirname
(
s
)
basename
=
os
.
path
.
basename
(
s
)
if
basename
.
endswith
(
'.py'
):
basename
=
basename
[:
-
3
]
newname
=
'tpk-'
+
basename
# install scripts with a prefix to avoid name confusion
# setup.py could be executed the second time in the same dir
if
not
os
.
path
.
isfile
(
newname
):
shutil
.
move
(
s
,
newname
)
scripts_to_install
.
append
(
newname
)
setup
(
setup
(
version
=
__version__
,
install_requires
=
req
,
install_requires
=
req
,
extras_require
=
{
extras_require
=
{
'all'
:
extra_req
'all'
:
extra_req
},
},
scripts
=
scripts_to_install
)
)
tensorpack/__init__.py
View file @
51239e9c
...
@@ -2,7 +2,6 @@
...
@@ -2,7 +2,6 @@
# File: __init__.py
# File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
__version__
=
'0.1'
from
tensorpack.train
import
*
from
tensorpack.train
import
*
from
tensorpack.models
import
*
from
tensorpack.models
import
*
...
@@ -11,3 +10,4 @@ from tensorpack.tfutils import *
...
@@ -11,3 +10,4 @@ from tensorpack.tfutils import *
from
tensorpack.callbacks
import
*
from
tensorpack.callbacks
import
*
from
tensorpack.dataflow
import
*
from
tensorpack.dataflow
import
*
from
tensorpack.predict
import
*
from
tensorpack.predict
import
*
from
tensorpack.libinfo
import
__version__
tox.ini
View file @
51239e9c
...
@@ -2,6 +2,7 @@
...
@@ -2,6 +2,7 @@
max-line-length
=
120
max-line-length
=
120
exclude
=
.git,
exclude
=
.git,
tensorpack/__init__.py,
tensorpack/__init__.py,
setup.py,
snippet,
snippet,
docs,
docs,
examples,
examples,
...
...
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