Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
bob
conda
Commits
029bb41b
Commit
029bb41b
authored
Aug 26, 2019
by
Amir MOHAMMADI
Browse files
[bob-devel] More tests for pytorch
parent
0c1c1c67
Pipeline
#32750
failed with stage
in 39 minutes and 23 seconds
Changes
3
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
conda/beat-devel/meta.yaml
View file @
029bb41b
...
...
@@ -2,7 +2,7 @@
{
%
set nopin =
[
'
ca-certificates'
,
'
certifi'
,
'
openssl'
,
'
libstdcxx-ng'
,
'
libgcc-ng'
,
'
libcxx'
,
'
libcxxabi'
]
%
}
package
:
name
:
beat-devel
version
:
2019.08.
14
version
:
2019.08.
26
build
:
# the build number must be zero all the time.
...
...
@@ -15,7 +15,7 @@ build:
requirements
:
host
:
-
python {{ python }}
-
bob-devel 2019.08.
14
-
bob-devel 2019.08.
26
-
docker-py 4.0.2
-
flask 1.0.2
-
flask-cors 3.0.7
...
...
conda/bob-devel/meta.yaml
View file @
029bb41b
...
...
@@ -4,7 +4,7 @@ package:
name
:
bob-devel
# please update the version of beat-devel and update bob-devel's version
# there as well in the same merge request.
version
:
2019.08.
14
version
:
2019.08.
26
build
:
# the build number must be zero all the time.
...
...
@@ -94,11 +94,11 @@ requirements:
{
%
endfor %
}
test
:
imports
:
-
torch
# [linux]
requires
:
-
numpy
-
ffmpeg
-
pytorch
# [linux]
-
torchvision
# [linux]
commands
:
# we expect these features from ffmpeg:
-
ffmpeg -codecs | grep "DEVI.S zlib"
# [unix]
...
...
conda/bob-devel/run_test.py
0 → 100644
View file @
029bb41b
import
sys
# couple of imports to see if packages are working
import
numpy
def
test_pytorch
():
import
torch
from
torchvision.models
import
DenseNet
model
=
DenseNet
()
t
=
torch
.
randn
(
1
,
3
,
224
,
224
)
out
=
model
(
t
)
assert
out
.
shape
[
1
]
==
1000
if
sys
.
platform
.
startswith
(
"linux"
):
# test if pytorch installation is sane
test_pytorch
()
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