import torch
class Args:
    def __init__(self):
        self.method = 'origin'
        self.device  = torch.device("cuda" if torch.cuda.is_available() else "cpu")
        self.dataset = 'CAVE' # self.dataset = 'chikusei'  # 'chikusei' or 'xiongan'
        self.pretrained_model_path = None
        self.batch_size = 32
        self.end_epoch = 150
        self.init_lr = 2e-4
        self.ratio = 32
        self.ker_size = self.ratio # 往往模糊核大小 ker_size 与 降采样率 ratio 保持一致
        self.sigma = 2
        self.patch_size = 128
        self.stride = 6
        self.data_path = f'/yehui/datasets/{self.dataset}/'
        self.gen_path = f'/yehui/datasets/'
        # self.PAN_path = '/yehui/datasets/QuickBird/PAN_1024/'
        # self.MS_path = '/yehui/datasets/QuickBird/MS_256/'
        # self.data_path = '/yehui/GuidedNet/dataset/PaviaU/PaviaU.mat'
        # self.gen_path = '/yehui/GuidedNet/dataset/'
        self.gpu_id = '1'
        self.iteration_batch = 100
        self.num_workers = 0
        self.hsi_channels = 4
        self.pan_channels = 1
        self.msi_channels = 4
        self.seed = 42
        self.auto_patch = True
        self.enable_log = True # Enable or disable logging
        self.log_path = f'/yehui/ResNet/logs/{self.dataset}/{self.method}/'
        self.result_path = f'/yehui/ResNet/results/{self.dataset}/{self.method}/'
        self.pretrained_model_path = f'/yehui/ResNet/logs/{self.dataset}/{self.method}/2025_07_28_14_53_22/model/net_149epoch.pth'