Torchscript Example, For instance, if you define a layer in forward instead of __init__, JIT will get confused.


Powered By GrowthZone