caffe迁移训练

Thadea ·
更新时间:2024-11-14
· 860 次阅读

1.断点续训 #!/usr/bin/env sh ROOT=/home/felix/Felix/caffe-augmentation-master/data/General LOG=$ROOT/models/densenet/train-'data +%Y-%m-%d-%H-%M-%S'.log CAFFE=./build/tools/caffe.bin $CAFFE train --solver=$ROOT/solver.prototxt --snapshot=$ROOT/models/densenet/_iter_28000.solverstate --gpu=all 2>&1 | tee $LOG 2.迁移学习(预训练模型) #!/usr/bin/env sh ROOT=/home/felix/Felix/caffe-augmentation-master/data/General LOG=$ROOT/models/densenet/train-'data +%Y-%m-%d-%H-%M-%S'.log CAFFE=./build/tools/caffe.bin $CAFFE train --solver=$ROOT/solver.prototxt --weights=$ROOT/models/densenet/_iter_28000.caffemodel --gpu=all 2>&1 | tee $LOG

参考文献:

https://blog.csdn.net/hjxu2016/article/details/70946008
作者:CPFelix



caffe 训练

需要 登录 后方可回复, 如果你还没有账号请 注册新账号
相关文章