From 35dc86f8476250f693aa419f404b2b4c8c381d92 Mon Sep 17 00:00:00 2001 From: tslil Date: Sat, 27 Mar 2021 23:59:33 -0400 Subject: Playing around with huge nn's, poorly trained. Not great at shallow depths --- include/weights.h | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) (limited to 'include/weights.h') diff --git a/include/weights.h b/include/weights.h index f736198..40e2161 100644 --- a/include/weights.h +++ b/include/weights.h @@ -12,16 +12,16 @@ General Public License for more details. You should have received a copy of the GNU General Public License - along with Takwrap. If not, see . + along with ct. If not, see . */ #define KERN_SIZE 3 -#define KERN_CHAN 6 -#define KERN_NUM 12 -#define DENSE1_NUM 11 -#define DENSE2_NUM 8 +#define KERN_CHAN 7 +#define KERN_NUM 64 +#define DENSE1_NUM 64 +#define DENSE2_NUM 32 -#define KERN_OSIZE (5-KERN_SIZE+1) // 5 +#define KERN_OSIZE (6-KERN_SIZE+1) // 5 #define CONV_NUM (KERN_NUM * KERN_OSIZE * KERN_OSIZE) // 108 extern const float conv2d_weights[KERN_NUM][KERN_SIZE][KERN_SIZE][KERN_CHAN]; -- cgit v1.2.3