Hugging Face's logo

Spaces:
not-lain
/
anime-seg
Runtime error

App Files Community
not-lain commited on
Commit
bce4a86
·
verified ·
1 Parent(s): 13e108d

Update train.py

Browse files
Files changed (1) hide show
  1. train.py +0 -1
train.py CHANGED
@@ -10,7 +10,6 @@ from pytorch_lightning.callbacks import ModelCheckpoint
10
  from torch.utils.data import DataLoader
11
  from huggingface_hub import PyTorchModelHubMixin
12
 
13
- from data_loader import create_training_datasets
14
  from model import ISNetDIS, ISNetGTEncoder, U2NET, U2NET_full2, U2NET_lite2, MODNet \
15
  , InSPyReNet, InSPyReNet_Res2Net50, InSPyReNet_SwinB
16
 
 
10
  from torch.utils.data import DataLoader
11
  from huggingface_hub import PyTorchModelHubMixin
12
 
 
13
  from model import ISNetDIS, ISNetGTEncoder, U2NET, U2NET_full2, U2NET_lite2, MODNet \
14
  , InSPyReNet, InSPyReNet_Res2Net50, InSPyReNet_SwinB
15