Python LinearDataset Examples

Example #1

def data_creator(config):
    """Constructs torch.utils.data.Dataset objects.

    Note that even though two Dataset objects can be returned,
    only one dataset will be used for training.

    Args:
        config: Configuration dictionary passed into ``PyTorchTrainer``

    Returns:
        One or Two Dataset objects. If only one Dataset object is provided,
        ``trainer.validate()`` will throw a ValueError.
    """
    return LinearDataset(2, 5), LinearDataset(2, 5, size=400)

Example #2

File: test_pytorch.py Project: kseager/ray

 def single_loader(config):
     return LinearDataset(2, 5, size=1000000)