Toggle navigation
Toggle navigation
This project
Loading...
Sign in
Carlos-Francisco Méndez-Cruz
/
curso-alianza-2024
Go to a project
Toggle navigation
Toggle navigation pinning
Projects
Groups
Snippets
Help
Project
Activity
Repository
Pipelines
Graphs
Issues
0
Merge Requests
0
Wiki
Snippets
Network
Create a new issue
Builds
Commits
Issue Boards
Authored by
cmendezc
2024-10-14 21:51:59 -0600
Browse Files
Options
Browse Files
Download
Email Patches
Plain Diff
Commit
cd2dc1bbc5ffa17f7ea22fab20334642a9c61904
cd2dc1bb
1 parent
43907810
NPU examples
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
1 deletions
ejemplo-npu/mnist_npu_example_v2.py
ejemplo-npu/mnist_npu_example_v2.py
View file @
cd2dc1b
...
...
@@ -8,7 +8,6 @@ import torch.optim as optim
from
torchvision
import
datasets
,
transforms
from
torch.optim.lr_scheduler
import
StepLR
class
Net
(
nn
.
Module
):
def
__init__
(
self
):
super
(
Net
,
self
)
.
__init__
()
...
...
@@ -114,6 +113,7 @@ def main():
train_kwargs
.
update
(
cuda_kwargs
)
test_kwargs
.
update
(
cuda_kwargs
)
# CFMC: Added to fix donwload error
datasets
.
MNIST
.
resources
=
[
(
'https://ossci-datasets.s3.amazonaws.com/mnist/train-images-idx3-ubyte.gz'
,
'f68b3c2dcbeaaa9fbdd348bbdeb94873'
),
...
...
@@ -122,6 +122,7 @@ def main():
(
'https://ossci-datasets.s3.amazonaws.com/mnist/t10k-images-idx3-ubyte.gz'
,
'9fb629c4189551a2d022fa330f9573f3'
),
(
'https://ossci-datasets.s3.amazonaws.com/mnist/t10k-labels-idx1-ubyte.gz'
,
'ec29112dd5afa0611ce80d1b7f02629c'
)
]
# End CFMC
transform
=
transforms
.
Compose
([
transforms
.
ToTensor
(),
...
...
Please
register
or
login
to post a comment