def test_dcgan(start_client_server_2_cpus):
assert ray.util.client.ray.is_connected()
from ray.util.sgd.torch.examples.dcgan import train_example
trainer = train_example(num_workers=2, use_gpu=False, test_mode=True)
_ = trainer.get_model()
def test_tf_train_example(start_client_server_2_cpus):
assert ray.util.client.ray.is_connected()
from ray.util.sgd.tf.examples.tensorflow_train_example import train_example
train_example(num_replicas=2, use_gpu=False)
def test_train_example(start_client_server_2_cpus):
assert ray.util.client.ray.is_connected()
from ray.util.sgd.torch.examples.train_example import train_example
train_example(num_workers=2, use_gpu=False)