(model, input, output)
| 1109 | activation = {} |
| 1110 | def getActivation(name): |
| 1111 | def hook(model, input, output): |
| 1112 | if name in activation: |
| 1113 | activation[name].append(output.detach()) |
| 1114 | else: |
| 1115 | activation[name] = [output.detach()] |
| 1116 | return hook |
| 1117 | |
| 1118 | if args.model == 'TGSynergy': |
nothing calls this directly
no outgoing calls
no test coverage detected