diff options
Diffstat (limited to 'test/rgb_part_net.py')
| -rw-r--r-- | test/rgb_part_net.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/test/rgb_part_net.py b/test/rgb_part_net.py index 1d754a0..d0d4e91 100644 --- a/test/rgb_part_net.py +++ b/test/rgb_part_net.py @@ -1,6 +1,6 @@ import torch -from models import RGBPartNet +from models.rgb_part_net import RGBPartNet P, K = 2, 4 N, T, C, H, W = P * K, 10, 3, 64, 32 |