def create_iterators()

in source/sagemaker/src/package/data_privatization/container/train.py [0:0]

• 9 lines of code
• 2 McCabe index (conditional complexity)

def create_iterators(train_data, valid_data):
    # Create iterators
    BATCH_SIZE = 64

    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

    train_iterator, valid_iterator = data.BucketIterator.splits(
        (train_data, valid_data),
        batch_size=BATCH_SIZE,
        sort=False,
        device=device)

    return train_iterator, valid_iterator