-
Notifications
You must be signed in to change notification settings - Fork 3.5k
Description
I want finetune model, and when i use trainer.ipynb i get this error `ImportError Traceback (most recent call last)
Cell In[1], line 4
2 import torch.backends.cudnn as cudnn
3 import yaml
----> 4 from train import train
5 from utils import AttrDict
6 import pandas as pd
File ~/pet/pcb/EasyOCR/trainer/train.py:15
12 import numpy as np
14 from utils import CTCLabelConverter, AttnLabelConverter, Averager
---> 15 from dataset import hierarchical_dataset, AlignCollate, Batch_Balanced_Dataset
16 from model import Model
17 from test import validation
File ~/pet/pcb/EasyOCR/trainer/dataset.py:13
11 import numpy as np
12 from torch.utils.data import Dataset, ConcatDataset, Subset
---> 13 from torch._utils import _accumulate
14 import torchvision.transforms as transforms
16 def contrast_grey(img):
ImportError: cannot import name '_accumulate' from 'torch._utils' (/home/simba9/anaconda3/envs/easyocr/lib/python3.9/site-packages/torch/_utils.py)`