Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
# SPDX-FileCopyrightText: Copyright © 2024 Idiap Research Institute <contact@idiap.ch>
#
# SPDX-License-Identifier: GPL-3.0-or-later
digraph G {
rankdir = T;
fontname = "Helvetica"
node [
fontname = "Helvetica"
shape = "record"
]
edge [
fontname = "Helvetica"
]
Database [
label = "Database\l(on storage)"
shape = "cylinder"
]
DatabaseSplit [
label = "{DatabaseSplit|+ __init__(description: JSON)\l+ splits() : dict[str, list]\l}"
]
RawDataLoader [
label = "{RawDataLoader|+ datadir : path\l|+ sample(description : JSON) : Sample \l+ label(description : JSON) : int\l}"
]
DataModule [
label = "{DataModule|- datasets : dict[str, torch.Dataset]\l+ model_transforms : TransformSequence\l|+ setup(stage: str)\l+ train_dataloader() : DataLoader\l+ val_dataloader() : dict[str, DataLoader]\l+ test_dataloader() : dict[str, DataLoader]\l+ predict_dataloader() : dict[str, DataLoader]\l}"
]
CachingDataModule [
label = "{CachingDataModule (lightning.DataModule)}"
style = "dashed"
]
Sample [
label = "{Sample (tuple)|+ tensor: torch.Tensor\l+ metadata: dict[str, Any]\l}"
]
DataLoader [
label = "{DataLoader (torch.DataLoader)|+ __getitem__(key: int)\l+ __iter__()\l}"
]
edge [
arrowhead = "empty"
]
DataModule -> CachingDataModule
edge [
arrowhead = "diamond"
taillabel = "1..1"
]
DatabaseSplit -> DataModule
RawDataLoader -> DataModule
edge [
arrowhead = "diamond"
taillabel = "1..*"
]
Sample -> DataLoader
edge [
arrowhead = "none"
taillabel = ""
label = "generates"
]
DataModule -> DataLoader
edge [
arrowhead = "none"
headlabel = "1..1"
label = "reads"
]
RawDataLoader -> Database
{ rank = same; Database; CachingDataModule; Sample; }
{ rank = same; RawDataLoader; DatabaseSplit; DataLoader; }
}