saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-01 23:30:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add IO binding support, which allows CUDA training

Browse source
This commit is contained in:
Thiago Crepaldi 2020-11-16 16:38:33 -08:00
parent f13c2a61d5
commit 4d9267e102
1 changed files with 0 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
orttraining/orttraining/test/python/orttraining_test_ortmodule_mnist.py
View file

@ -1,5 +1,3 @@
import pdb
import argparse
import logging
import torch