集群情况下需在host端发布训练命令
此变量对应的是1.x版本的conf文件,源码为:
algo
: 联邦学习算法的名称,默认为 "fedavg"。model
: 要使用的模型,它可以是 PyTorch 的Module
、Sequential
类型或者某种模型加载器(ModelLoader
)。optimizer
: 优化器,它可以是 PyTorch 的优化器类型或者加载器(Loader
)。loss
: 损失函数,它可以是 PyTorch 的模块或者自定义函数的加载器(CustFuncLoader
)。training_args
: 训练参数,类型为TrainingArguments
。fed_args
: 联邦学习参数,类型为FedArguments
。dataset
: 数据集加载器(DatasetLoader
)。data_collator
: 数据整理函数加载器(CustFuncLoader
)。tokenizer
: 分词器加载器(CustFuncLoader
)。task_type
: 任务类型,如 "binary"(二分类)、"multi"(多分类)、"regression"(回归)等。