Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
class SaveOutput:
def __init__(self):
self.outputs = []
def __call__(self, module, module_in, module_out):
self.outputs.append(module_out)
def clear(self):
self.outputs = []
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment