aligner
Aligner
¶
Bases: BaseGCAdapter
Wraps AlignerPlusCHook.
Container | Required keys |
---|---|
models | ["G", "C"] |
optimizers | ["G", "C"] |
Source code in pytorch_adapt\adapters\aligner.py
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
|
RTN
¶
Bases: Aligner
Wraps RTNHook.
Container | Required keys |
---|---|
models | ["G", "C", "residual_model"] |
optimizers | ["G", "C", "residual_model"] |
misc | ["feature_combiner"] |
Source code in pytorch_adapt\adapters\aligner.py
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 |
|
__init__(*args, inference_fn=None, **kwargs)
¶
Parameters:
Name | Type | Description | Default |
---|---|---|---|
inference_fn |
Default is rtn_fn |
None
|
Source code in pytorch_adapt\adapters\aligner.py
39 40 41 42 43 44 45 |
|