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
c15c8639
Commit
c15c8639
authored
Mar 19, 2017
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
bump version. relax gpu requirements
parent
20c7fcb4
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
12 additions
and
9 deletions
+12
-9
.travis.yml
.travis.yml
+5
-5
tensorpack/libinfo.py
tensorpack/libinfo.py
+1
-1
tensorpack/train/multigpu.py
tensorpack/train/multigpu.py
+6
-3
No files found.
.travis.yml
View file @
c15c8639
...
...
@@ -16,18 +16,18 @@ matrix:
include
:
-
os
:
linux
python
:
2.7
env
:
TF_VERSION=1.0.
0rc2
TF_TYPE=release
env
:
TF_VERSION=1.0.
1
TF_TYPE=release
-
os
:
linux
python
:
3.5
env
:
TF_VERSION=1.0.
0rc2
TF_TYPE=release
env
:
TF_VERSION=1.0.
1
TF_TYPE=release
-
os
:
linux
python
:
2.7
env
:
TF_VERSION=1.0.
0rc2
TF_TYPE=nightly
env
:
TF_VERSION=1.0.
1
TF_TYPE=nightly
-
os
:
linux
python
:
3.5
env
:
TF_VERSION=1.0.
0rc2
TF_TYPE=nightly
env
:
TF_VERSION=1.0.
1
TF_TYPE=nightly
allow_failures
:
-
env
:
TF_VERSION=1.0.
0rc2
TF_TYPE=nightly
-
env
:
TF_VERSION=1.0.
1
TF_TYPE=nightly
install
:
-
pip install -U pip
# the pip version on travis is too old
...
...
tensorpack/libinfo.py
View file @
c15c8639
...
...
@@ -6,4 +6,4 @@ import cv2 # noqa
import
os
os
.
environ
[
'OPENCV_OPENCL_RUNTIME'
]
=
''
__version__
=
'0.1.
6
'
__version__
=
'0.1.
7
'
tensorpack/train/multigpu.py
View file @
c15c8639
...
...
@@ -96,8 +96,9 @@ class SyncMultiGPUTrainer(MultiGPUTrainer,
config
.
predict_tower
=
predict_tower
super
(
SyncMultiGPUTrainer
,
self
)
.
__init__
(
config
)
assert
len
(
config
.
tower
)
>=
1
,
"MultiGPUTrainer must be used with at least one GPU."
assert
tf
.
test
.
is_gpu_available
()
assert
len
(
config
.
tower
)
>=
1
,
"MultiGPUTrainer must be used with at least one tower."
if
len
(
config
.
tower
)
>
1
:
assert
tf
.
test
.
is_gpu_available
()
self
.
average_cost
=
average_cost
@
staticmethod
...
...
@@ -185,7 +186,9 @@ class AsyncMultiGPUTrainer(MultiGPUTrainer,
config
.
predict_tower
=
predict_tower
self
.
_scale_gradient
=
scale_gradient
assert
tf
.
test
.
is_gpu_available
()
if
len
(
config
.
tower
)
>
1
:
assert
tf
.
test
.
is_gpu_available
()
def
_setup
(
self
):
super
(
AsyncMultiGPUTrainer
,
self
)
.
_setup
()
...
...
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