Update body model
Browse files- body/Model.py +1 -0
body/Model.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
from dynamic_network_architectures.architectures.unet import ResidualEncoderUNet
|
| 2 |
from konfai.network import network
|
| 3 |
|
|
|
|
| 1 |
+
from torch import nn
|
| 2 |
from dynamic_network_architectures.architectures.unet import ResidualEncoderUNet
|
| 3 |
from konfai.network import network
|
| 4 |
|