# Unable to import 'weak\_module'

**URL:** <https://discuss.pytorch.org/t/unable-to-import-weak-module/60289>\
**Category:** jit\
**Created:** [November 7, 2019, 3:18pm UTC](https://discuss.pytorch.org/t/unable-to-import-weak-module/60289 "2019-11-07T15:18:40Z")\
**Posts on this page:** 8\
**Page:** 1

<div class="post-metadata">

**Author:** ![Soniya](https://discuss.pytorch.org/letter_avatar_proxy/v4/letter/s/5fc32e/32.png) [@Soniya](https://discuss.pytorch.org/u/Soniya)\
**Post date:** [November 7, 2019, 3:18pm UTC](https://discuss.pytorch.org/t/unable-to-import-weak-module/60289/1 "2019-11-07T15:18:40Z")

</div>

Hi Everyone,  
I’m unable to import ‘weak\_module’ from ‘torch.\_jit\_internal’. It says 'ImportError: cannot import name ‘weak\_module’. I’m using Pytorch 1.3.0a0+24ae9b5 version. Please help me.

---

<div class="post-metadata">

**Author:** ![albanD](https://discuss.pytorch.org/user_avatar/discuss.pytorch.org/alband/32/215_2.png) [@albanD](https://discuss.pytorch.org/u/albanD)\
**Post date:** [November 7, 2019, 3:23pm UTC](https://discuss.pytorch.org/t/unable-to-import-weak-module/60289/2 "2019-11-07T15:23:03Z")

</div>

This is some internal module, it might have been removed/renamed. Why do you want to use it?

---

<div class="post-metadata">

**Author:** ![Soniya](https://discuss.pytorch.org/letter_avatar_proxy/v4/letter/s/5fc32e/32.png) [@Soniya](https://discuss.pytorch.org/u/Soniya)\
**Post date:** [November 7, 2019, 3:28pm UTC](https://discuss.pytorch.org/t/unable-to-import-weak-module/60289/3 "2019-11-07T15:28:48Z")

</div>

I’m trying to run EfficientNet which has used the module in the following manner.  
‘from torch.\_jit\_internal import weak\_module, weak\_script\_method’

---

<div class="post-metadata">

**Author:** ![driazati](https://discuss.pytorch.org/user_avatar/discuss.pytorch.org/driazati/32/15393_2.png) [@driazati](https://discuss.pytorch.org/u/driazati)\
**Post date:** [November 7, 2019, 5:06pm UTC](https://discuss.pytorch.org/t/unable-to-import-weak-module/60289/4 "2019-11-07T17:06:42Z")

</div>

> [@Soniya](#):
>
> EfficientNet

Which EfficientNet implementation are you using? The decorators its pulling from `_jit_internal` were deleted in v1.2. Their functionality is now automatic, so they can just be deleted and everything should work the same.

---

<div class="post-metadata">

**Author:** ![Soniya](https://discuss.pytorch.org/letter_avatar_proxy/v4/letter/s/5fc32e/32.png) [@Soniya](https://discuss.pytorch.org/u/Soniya)\
**Post date:** [November 7, 2019, 5:26pm UTC](https://discuss.pytorch.org/t/unable-to-import-weak-module/60289/5 "2019-11-07T17:26:04Z")

</div>

I’m using the implementation which is given here [https://github.com/katsura-jp/efficientnet-pytorch](https://github.com/katsura-jp/efficientnet-pytorch)

---

<div class="post-metadata">

**Author:** ![driazati](https://discuss.pytorch.org/user_avatar/discuss.pytorch.org/driazati/32/15393_2.png) [@driazati](https://discuss.pytorch.org/u/driazati)\
**Post date:** [November 7, 2019, 5:32pm UTC](https://discuss.pytorch.org/t/unable-to-import-weak-module/60289/6 "2019-11-07T17:32:54Z")

</div>

It should work if you apply this patch

```diff
diff --git a/model/swish.py b/model/swish.py
index 66adfa5..3f68678 100644
--- a/model/swish.py
+++ b/model/swish.py
@@ -1,10 +1,8 @@
 import torch
 import torch.nn as nn
 from torch.nn.parameter import Parameter
-from torch._jit_internal import weak_module, weak_script_method
 
 
-@weak_module
 class Swish(nn.Module):
     def __init__ (self, train_beta=False):
         super(Swish, self). __init__ ()
@@ -13,7 +11,6 @@ class Swish(nn.Module):
         else:
             self.weight = 1.0
 
- @weak_script_method
     def forward(self, input):
         return input * torch.sigmoid(self.weight * input)

```

---

<div class="post-metadata">

**Author:** ![Soniya](https://discuss.pytorch.org/letter_avatar_proxy/v4/letter/s/5fc32e/32.png) [@Soniya](https://discuss.pytorch.org/u/Soniya)\
**Post date:** [November 8, 2019, 12:22pm UTC](https://discuss.pytorch.org/t/unable-to-import-weak-module/60289/7 "2019-11-08T12:22:52Z")

</div>

Where to apply this patch? I’m unable to find the corresponding file from where i have used it as ‘from torch.\_jit\_internal’.

---

<div class="post-metadata">

**Author:** ![albanD](https://discuss.pytorch.org/user_avatar/discuss.pytorch.org/alband/32/215_2.png) [@albanD](https://discuss.pytorch.org/u/albanD)\
**Post date:** [November 8, 2019, 3:39pm UTC](https://discuss.pytorch.org/t/unable-to-import-weak-module/60289/8 "2019-11-08T15:39:35Z")

</div>

Hi,

He refers to [this file](https://github.com/katsura-jp/efficientnet-pytorch/blob/master/model/swish.py) in the repo you linked above.