【笔记00001】pytorch 训练模型-初始化需要的参数

无限添加需要的参数

#初始化参数
class Parser:
    def __init__(self):
        self.epochs = 2
        self.lr = 0.000001
        self.test_batch_size = 8
        self.batch_size = 32
        self.log_interval = 10
        self.seed = 1

        self.workerNum = 5          
        self.iterations = 10          
        self.worker_iters = 3         

        self.SgdLr = 0.1


args = Parser()
torch.manual_seed(args.seed)