File size: 56 Bytes
5fa1a76
1
This feature can be used with any nn.Module-based model.