使用 JSON 文件配置神经网络

青葱年少 pytorch 317

原文标题 ：config a neural network with a JSON file

我想从 JSON 文件中配置一个神经网络模型（层数、每层神经元数、激活函数……）。但老实说，我不知道该怎么做。当我用“ hyperparameters config PyTorch”或“hyperparameters tunning PyTorch”找不到任何有趣的东西。搜索结果更多的是关于超参数优化，而不是来自 JSON 的配置。任何人都知道如何做到这一点（JSON 文件配置）或者你知道任何有用的我可以观看/阅读的教程，好吗？那将是一个很大的帮助！
先感谢您

原文链接：https://stackoverflow.com//questions/71572817/config-a-neural-network-with-a-json-file

回复

我来回复
  • Ynjxsjmh的头像
    Ynjxsjmh 评论

    将您的参数写入具有正确名称的 json 文件，如下所示

    {
   "number_layers":1,
   "number_neurons":2,
   "activation_function":"relu",
   "training":{
      "learning_rate":0.01
   }
}

    然后读取json文件

    import json


with open('xxx.json', 'r', encoding='utf-8') as f:
    config = json.loads(f.read())

    并访问您想要的参数

    config['number_layers']
config['training']['learning_rate']
    2年前 0条评论