37f58cc580da8735ae5a259d4cb4ba0b8701e78e,tests/test_hogwild.py,,,#,16

Before Change

import torch


SKIP_HOGWILD = torch.cuda.device_count() > 0


class display_output(object):
    def __init__(self):

After Change

NUM_EXS = 100
// ideally we want one choice which is a nice modulo with NUM_EXS, and one that isn"t
NUM_THREADS_CHOICES = [2, 8]
BATCHSIZE_CHOICES = [1, 8]


@testing_utils.skipIfGPU
class TestHogwild(unittest.TestCase):

Instances