andr3/gotch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore(example/cifar): switch to GPU for colab test

Browse Source
This commit is contained in:
sugarme 2020-07-08 20:20:53 +10:00
parent 787734f624
commit bb4e286ee5
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
example/cifar/main.go
View File

@ -107,9 +107,9 @@ func main() {
startRAM := si.TotalRam - si.FreeRam
// cuda := gotch.CudaBuilder(0)
// device := cuda.CudaIfAvailable()
device := gotch.CPU
cuda := gotch.CudaBuilder(0)
device := cuda.CudaIfAvailable()
// device := gotch.CPU
vs := nn.NewVarStore(device)