Base for nn conversion

2025-10-20 21:14:14 +08:00 · 2016-07-22 14:27:00 -04:00
parent 554a1d8336
commit ae40bcd58c
10 changed files with 1648 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@ -83,5 +83,5 @@ C = Extension("torch._C",

 setup(name="torch", version="0.1",
      ext_modules=[C],
-      packages=['torch', 'torch.cuda', 'torch.optim'],
+      packages=['torch', 'torch.cuda', 'torch.optim', 'torch.legacy', 'torch.legacy.nn'],
 )