Browse Source

requirements updated

master
Faeze 1 year ago
parent
commit
7f7feeb00b
4 changed files with 7 additions and 8 deletions
  1. 2
    1
      .gitignore
  2. 1
    2
      data/twitter/data_loader.py
  3. 0
    3
      main.py
  4. 4
    2
      requirements.txt

+ 2
- 1
.gitignore View File

@@ -121,4 +121,5 @@ results/
*.jpg
*.csv
*.json
*.png
*.png
*.zip

+ 1
- 2
data/twitter/data_loader.py View File

@@ -1,7 +1,6 @@

import cv2
import pickle
from PIL import Image
from cv2 import cv2
from numpy import asarray

from data.data_loader import DatasetLoader

+ 0
- 3
main.py View File

@@ -1,6 +1,5 @@
import argparse

from lime_main import lime_main
from optuna_main import optuna_main
from test_main import test_main
from torch_main import torch_main
@@ -52,8 +51,6 @@ if __name__ == '__main__':
optuna_main(config, args.use_optuna)
elif args.just_test is not None:
test_main(config, args.just_test)
elif args.use_lime is not None:
lime_main(config, args.use_lime)
else:
torch_main(config)


+ 4
- 2
requirements.txt View File

@@ -5,7 +5,6 @@ Pillow==8.2.0
pytorch-lightning==1.2.10
pytorch-model-summary==0.1.2
pytorchtools==0.0.2
scikit-image==0.17.2
scikit-learn==0.24.2
scipy==1.5.4
seaborn==0.11.1
@@ -14,4 +13,7 @@ torchmetrics==0.2.0
torchsummary==1.5.1
torchvision==0.9.1
tqdm==4.60.0
transformers==4.5.1
tokenizers==0.13.3
transformers==4.30.0
timm==0.4.5
albumentations==0.5.2

Loading…
Cancel
Save