From 13209df7bb53de19aa3f82e870db11eb5b7587f1 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 13 May 2016 18:59:43 +0000
Subject: [PATCH] art, cudnn
---
src/network.c | 2 ++
1 files changed, 2 insertions(+), 0 deletions(-)
diff --git a/src/network.c b/src/network.c
index ca485d6..8f39f7b 100644
--- a/src/network.c
+++ b/src/network.c
@@ -65,6 +65,8 @@
return net.learning_rate * pow(net.gamma, batch_num);
case POLY:
return net.learning_rate * pow(1 - (float)batch_num / net.max_batches, net.power);
+ case RANDOM:
+ return net.learning_rate * pow(rand_uniform(0,1), net.power);
case SIG:
return net.learning_rate * (1./(1.+exp(net.gamma*(batch_num - net.step))));
default:
--
Gitblit v1.10.0