文件名称:前列腺癌段
文件大小:91KB
文件格式:ZIP
更新时间:2024-03-23 15:55:00
Python
暴风雨 该存储库提供了用于训练神经网络的管道。 使用的主要框架: 管道的主要思想: 所有参数和模块都在配置中定义; 事先为不同的优化器/调度器等准备配置,因此很容易在它们之间进行切换; 有用于不同深度学习任务的模板。 当前,支持图像分类和命名实体识别。 运行管道的示例:这将在MNIST上进行训练(将下载数据): >>> python train.py --config-name mnist_config model.encoder.params.to_one_channel=True 默认运行: >>> python train.py 管道的默认版本在imagenette数据集上运行。 为此,请从此存储库下载数据: : 解压缩并在conf / datamodule / image_classification.yaml路径中定义其路径。