From 4d9267e102b3ab6b055a3c87dc0ef238bb97fac5 Mon Sep 17 00:00:00 2001 From: Thiago Crepaldi Date: Mon, 16 Nov 2020 16:38:33 -0800 Subject: [PATCH] Add IO binding support, which allows CUDA training --- .../orttraining/test/python/orttraining_test_ortmodule_mnist.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/orttraining/orttraining/test/python/orttraining_test_ortmodule_mnist.py b/orttraining/orttraining/test/python/orttraining_test_ortmodule_mnist.py index 9147d981d7..aed5a59d1e 100644 --- a/orttraining/orttraining/test/python/orttraining_test_ortmodule_mnist.py +++ b/orttraining/orttraining/test/python/orttraining_test_ortmodule_mnist.py @@ -1,5 +1,3 @@ -import pdb - import argparse import logging import torch