Thank you!  A quick followup - on PyTorch versions before 1.1.0, is there an alternative for torch.nn.Identity.  I want to replace some of these fc layers with dummies if possible.  Thank you! I am constrained to using an older version of torch (1.0.2)