This website works better with JavaScript.
Home
Help
Explore
${searchName}
Repositories
${searchName}
Users
${searchName}
Organizations
English
English
简体中文
Sign In
family
/
AI-1
forked from
chenwenbin/AI-1
Watch
2
Star
1
Fork
0
Code
Issues
0
Pull Requests
0
Wiki
Activity
Browse Source
更新 'script.py'
master
chenwenbin
4 years ago
parent
f19c7e4bdf
commit
5951217130
1 changed files
with
1 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+1
-1
script.py
+ 1
- 1
script.py
View File
@ -48,7 +48,7 @@ class Model(nn.Module):
model
=
Model
(
input_size
,
output_size
)
model
=
nn
.
DataParallel
(
model
,
device_ids
=
[
0
,
1
,
2
,
3
]
)
.
cuda
(
)
model
.
to
(
device
)
#model.to(device
)
for
data
in
rand_loader
:
if
torch
.
cuda
.
is_available
(
)
:
Write
Preview
Loading…
Cancel
Save