logo
Browse Source

Fix load trainer issue

Signed-off-by: Kaiyuan Hu <kaiyuan.hu@zilliz.com>
main
Kaiyuan Hu 2 years ago
parent
commit
e5905a8d66
  1. 5
      isc.py

5
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

Loading…
Cancel
Save