This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
Register
Sign In
enmg4a
/
tw
forked from
kyxt/gpunpu
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
0
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
Browse Source
更新 'gpu/train.py'
test_v20221116
wjtest1201
2 years ago
parent
d430e901ca
commit
7d2e3dbf6f
1 changed files
with
1 additions
and
0 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+1
-0
gpu/train.py
+ 1
- 0
gpu/train.py
View File
@@ -30,6 +30,7 @@ from torch.optim import SGD
from torch.utils.data import DataLoader
from torchvision.transforms import ToTensor
import argparse
import datetime
# Training settings
parser = argparse.ArgumentParser(description='PyTorch MNIST Example')
Write
Preview
Loading…
Cancel
Save