1)defults.py中训练和验证集文件夹路径改为绝对路径
2)darkflow/net/help.py:
def build_train_op(self): | ||
self.framework.loss(self.out) | ||
self.say('Building {} train op'.format(self.meta['model'])) | ||
- optimizer = self._TRAINER[self.FLAGS.trainer](self.FLAGS.lr) | ||
+ self.learning_rate = tf.placeholder(tf.float32, shape=[]) | ||
+ optimizer = self._TRAINER[self.FLAGS.trainer](self.learning_rate) | ||
gradients = optimizer.compute_gradients(self.framework.loss) | ||
self.train_op = optimizer.apply_gradients(gradients) |
没有评论:
发表评论