From 257c7a5ad91a1d60d235059fb59a11e1915af9ae Mon Sep 17 00:00:00 2001 From: borg323 Date: Tue, 21 Feb 2023 15:13:10 +0200 Subject: [PATCH] add float16 encoding --- proto/net.proto | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/proto/net.proto b/proto/net.proto index c52f6ca..c8649e9 100644 --- a/proto/net.proto +++ b/proto/net.proto @@ -92,6 +92,13 @@ message Weights { optional Layer ln2_betas = 6; } + enum Encoding { + UNKNOWN = 0; + LINEAR16 = 1; + FLOAT16 = 2; + } + optional Encoding encoding = 37; + // Input convnet. optional ConvBlock input = 1; @@ -211,7 +218,7 @@ message Format { UNKNOWN = 0; LINEAR16 = 1; } - + // Here for comatibility with old nets, the correct one is in Weights. optional Encoding weights_encoding = 1; // If network_format is missing, it's assumed to have // INPUT_CLASSICAL_112_PLANE / OUTPUT_CLASSICAL / NETWORK_CLASSICAL format.