cuda panics on batches larger than 1024 so skip those and fallback to cpu |
||
|---|---|---|
| .. | ||
| backend | ||
| nn | ||
| backend.go | ||
| device.go | ||
| path.go | ||
cuda panics on batches larger than 1024 so skip those and fallback to cpu |
||
|---|---|---|
| .. | ||
| backend | ||
| nn | ||
| backend.go | ||
| device.go | ||
| path.go | ||