def forward(self, x):
with annotate(aaa='a', bbb=['b', 'c']):
h = self.conv(x)
h = self.conv2(h)
with annotate(zzz=99, yyy=[9, 9]):
h = self.linear(h)
h = self.linear2(h)
return h
def forward(self, x):
with annotate(aaa='a'):
h = self.conv(x)
h = self.linear(h)
return h