Ejemplos de GPUStatsMonitor._parse_gpu_stats en Python

Ejemplo n.º 1

Archivo: test_gpu_stats_monitor.py Proyecto: oddeirikigland/pytorch-lightning

def test_gpu_stats_monitor_parse_gpu_stats():
    logs = GPUStatsMonitor._parse_gpu_stats([1, 2], [[3, 4, 5], [6, 7]], [("gpu", "a"), ("memory", "b")])
    expected = {
        "device_id: 1/gpu (a)": 3,
        "device_id: 1/memory (b)": 4,
        "device_id: 2/gpu (a)": 6,
        "device_id: 2/memory (b)": 7,
    }
    assert logs == expected

Ejemplo n.º 2

Archivo: test_gpu_stats_monitor.py Proyecto: biswaroop1547/pytorch-lightning

def test_gpu_stats_monitor_parse_gpu_stats():
    logs = GPUStatsMonitor._parse_gpu_stats('1,2', [[3, 4, 5], [6, 7]],
                                            [('gpu', 'a'), ('memory', 'b')])
    expected = {
        'gpu_id: 1/gpu (a)': 3,
        'gpu_id: 1/memory (b)': 4,
        'gpu_id: 2/gpu (a)': 6,
        'gpu_id: 2/memory (b)': 7
    }
    assert logs == expected