Keras-Project-Template

Keras 项目模板 一个项目模板,用于简化使用 Keras 构建和训练 深度学习 模型。

入门

此模板允许您简单地构建和训练带有检查点和张量板可视化的 深度学习 模型。 为了使用此模板,您必须: 1. 定义一个数据加载器类。 2. 定义一个继承自 BaseModel 的模型类。 3. 定义一个继承自 BaseTrainer 的训练器类。 4. 定义一个配置文件,其中包含实验所需的参数。 5. 使用以下命令运行模型: shell python main.py -c [path to configuration file] # 运行示例项目 为 mnist 数据集提供了一个简单的模型,可用于测试此模板。 要运行示例项目,请执行以下操作: 1. 使用以下命令启动训练: shell python main.py -c configs/simple_mnist_config.json 2. 使用以下命令启动张量板可视化: ```shell tensorboard --logdir=experiments/simple_mnist/logs

# Comet.ml 集成

### 模型
您需要:
1. 创建一个继承自 **BaseModel** 的模型类。
2. 重写定义模型的 ***build_model*** 函数。
3. 在构造函数中调用 ***build_model*** 函数。