- import torch
- DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
- DATA_PATH1 = r"C:\Programming\Datasets\Fake News Ru\train_bodies.csv"
- DATA_PATH2 = r"C:\Programming\Datasets\Fake News Ru\train_stances.csv"
- CHECKPOINT_PATH = r"TextClassifictaion.tar"
- INP_SEQ_LEN = 900
- LR_RATE = 1e-4
- WEIGHT_DECAY = 1e-4
- BATCH_SIZE = 10
- NUM_WORKERS = 0
|
