dataloader_creator
DataloaderCreator
¶
This is a factory class for creating dataloaders.
The __call__
function takes in keyword arguments which are datasets,
and outputs a dictionary of dataloaders (one dataloader for each input dataset).
Source code in pytorch_adapt\datasets\dataloader_creator.py
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 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 |
|
__call__(**kwargs)
¶
Parameters:
Name | Type | Description | Default |
---|---|---|---|
**kwargs |
keyword arguments mapping from dataset names to datasets. |
{}
|
Returns:
Type | Description |
---|---|
Dict[str, DataLoader]
|
a dictionary mapping from dataset names to dataloaders. |
Source code in pytorch_adapt\datasets\dataloader_creator.py
96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 |
|
__init__(train_kwargs=None, val_kwargs=None, train_names=None, val_names=None, all_train=False, all_val=False, batch_size=32, num_workers=0)
¶
Parameters:
Name | Type | Description | Default |
---|---|---|---|
train_kwargs |
Dict[str, Any]
|
The keyword arguments that will be
passed to every DataLoader constructor for train-time datasets.
If
|
None
|
val_kwargs |
Dict[str, Any]
|
The keyword arguments that will be
passed to every DataLoader constructor for validation-time datasets.
If
|
None
|
train_names |
List[str]
|
A list of the dataset names that are used during training.
If |
None
|
val_names |
List[str]
|
A list of the dataset names that are used during validation.
If |
None
|
all_train |
bool
|
If True, then all input datasets are assumed to be for training, regardless of their names. |
False
|
all_val |
bool
|
If True, then all input datasets are assumed to be for validation, regardless of their names. |
False
|
batch_size |
int
|
The |
32
|
num_workers |
int
|
The |
0
|
Source code in pytorch_adapt\datasets\dataloader_creator.py
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 90 91 92 93 94 |
|