aboutsummaryrefslogtreecommitdiff
path: root/include/weights.h
diff options
context:
space:
mode:
authortslil <tslil@posteo.de>2021-03-27 23:59:33 -0400
committertslil <tslil@posteo.de>2026-08-28 19:37:41 +0100
commit35dc86f8476250f693aa419f404b2b4c8c381d92 (patch)
tree678cd42cdbf0e46662dd2e71694e6c496d8e2279 /include/weights.h
parent00a04c2929bdc8f8f4bf7d5d8cf413ebfb3cd006 (diff)
Playing around with huge nn's, poorly trained.sixes
Not great at shallow depths
Diffstat (limited to 'include/weights.h')
-rw-r--r--include/weights.h12
1 files changed, 6 insertions, 6 deletions
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 <https://www.gnu.org/licenses/>.
+ along with ct. If not, see <https://www.gnu.org/licenses/>.
*/
#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];