base
BaseConditionHook
¶
Bases: BaseHook
The base class for hooks that return a boolean
Source code in pytorch_adapt\hooks\base.py
176 177 178 179 180 181 182 183 184 185 |
|
BaseHook
¶
Bases: ABC
All hooks extend BaseHook
Source code in pytorch_adapt\hooks\base.py
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 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 |
|
__init__(loss_prefix='', loss_suffix='', out_prefix='', out_suffix='', key_map=None)
¶
Parameters:
Name | Type | Description | Default |
---|---|---|---|
loss_prefix |
str
|
prepended to all new loss keys |
''
|
loss_suffix |
str
|
appended to all new loss keys |
''
|
out_prefix |
str
|
prepended to all new output keys |
''
|
out_suffix |
str
|
appended to all new output keys |
''
|
key_map |
Dict[str, str]
|
a mapping from |
None
|
Source code in pytorch_adapt\hooks\base.py
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 |
|
call(inputs, losses)
abstractmethod
¶
This gets called by __call__
and must be implemented by the child class.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
inputs |
Dict[str, Any]
|
holds data and models |
required |
losses |
Dict[str, Any]
|
previously computed losses |
required |
Returns:
Type | Description |
---|---|
Union[Tuple[Dict[str, Any], Dict[str, Any]], bool]
|
Either a tuple of |
Source code in pytorch_adapt\hooks\base.py
74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
|
BaseWrapperHook
¶
Bases: BaseHook
A simple wrapper for calling self.hook
,
which should be defined in the child's __init__
function.
Source code in pytorch_adapt\hooks\base.py
188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 |
|