PaperText_Segmentation/Utils/train_configs.py

import torch

data_dir = r"Data/samples"
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
learning_rate = 1e-5  # 0.0003  # 0.0003 学习率  2e-5太小了 1e-4 
num_epochs = 1
patience = 10
model_dimension = 512