安装
克隆存储库
git clone git@github.com:awni/ecg.git
如果您没有virtualenv ,请使用
pip install virtualenv
制作并激活新的Python 2.7环境
virtualenv -p python2.7 ecg_env
source ecg_env/bin/activate
安装要求(这可能需要几分钟)。
对于仅CPU支持运行
./setup.sh
要安装GPU支持,请运行
env TF=gpu ./setup.sh
训练
在回购根目录( ecg )中,创建一个名为saved的新目录。
mkdir saved
要训练模型,请使用以下命令,将path_to_config.json替换为实际配置:
python ecg/train.py path_to_config.json
请注意,在每个时期之后,将模型保存在ecg
2021-11-08 11:09:46
266KB
Python
1