| |
|
| |
|
| |
|
| |
|
| |
|
| | import abc |
| |
|
| |
|
| | class NetModifier(metaclass=abc.ABCMeta): |
| | """ |
| | An abstraction class for supporting modifying a generated net. |
| | Inherited classes should implement the modify_net method where |
| | related operators are added to the net. |
| | |
| | Example usage: |
| | modifier = SomeNetModifier(opts) |
| | modifier(net) |
| | """ |
| |
|
| | def __init__(self): |
| | pass |
| |
|
| | @abc.abstractmethod |
| | def modify_net(self, net, init_net=None, grad_map=None, blob_to_device=None): |
| | pass |
| |
|
| | def __call__(self, net, init_net=None, grad_map=None, blob_to_device=None, |
| | modify_output_record=False): |
| | self.modify_net( |
| | net, |
| | init_net=init_net, |
| | grad_map=grad_map, |
| | blob_to_device=blob_to_device, |
| | modify_output_record=modify_output_record) |
| |
|
