Add new model to torch.legacy.nn

1

Hello,

I’m trying to add a new file to torch.legacy.nn but Ihv seen that all of this files are already compiled.
When I add a new python file it doesn’t effect anything.

for example if I do:

import torch.legacy.nn as nn
nn.dict # doesn’t show the new file that I added.

Thanks,
Tsachi

2

Hi,

You need to add to torch/legacy/nn/__init__.py your module the same way as the others.