Browse Source
Fix load trainer issue
Signed-off-by: Kaiyuan Hu <kaiyuan.hu@zilliz.com>
main
1 changed files with
3 additions and
2 deletions
-
isc.py
|
@ -14,6 +14,7 @@ |
|
|
|
|
|
|
|
|
import logging |
|
|
import logging |
|
|
import os |
|
|
import os |
|
|
|
|
|
import warnings |
|
|
from typing import Union, List |
|
|
from typing import Union, List |
|
|
from pathlib import Path |
|
|
from pathlib import Path |
|
|
|
|
|
|
|
@ -22,6 +23,7 @@ from towhee.operator.base import NNOperator, OperatorFlag |
|
|
from towhee.types.arg import arg, to_image_color |
|
|
from towhee.types.arg import arg, to_image_color |
|
|
from towhee import register |
|
|
from towhee import register |
|
|
from towhee.models import isc |
|
|
from towhee.models import isc |
|
|
|
|
|
import sys |
|
|
# from towhee.dc2 import accelerate |
|
|
# from towhee.dc2 import accelerate |
|
|
|
|
|
|
|
|
import torch |
|
|
import torch |
|
@ -30,10 +32,9 @@ from torchvision import transforms |
|
|
from PIL import Image as PILImage |
|
|
from PIL import Image as PILImage |
|
|
import timm |
|
|
import timm |
|
|
|
|
|
|
|
|
import warnings |
|
|
|
|
|
|
|
|
|
|
|
warnings.filterwarnings('ignore') |
|
|
warnings.filterwarnings('ignore') |
|
|
log = logging.getLogger('isc_op') |
|
|
log = logging.getLogger('isc_op') |
|
|
|
|
|
_ = sys.modules[__name__] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# @accelerate |
|
|
# @accelerate |
|
|