As I said, I’m not familiar with the DL4J implementation, so you would need to ask the authors in their discussion board of on GitHub etc.
If you think that the torchvision model doesn’t reflect the expected implementation from the paper, please let us know and we’ll look into it.