From 3cd8acb919e963a380c374a6ed6862e679769a75 Mon Sep 17 00:00:00 2001 From: Andre Henriques Date: Sat, 10 Feb 2024 09:41:16 +0000 Subject: [PATCH] fix: closes #63 --- logic/models/edit.go | 4 ++-- logic/models/train/train.go | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/logic/models/edit.go b/logic/models/edit.go index 256d270..63cb37f 100644 --- a/logic/models/edit.go +++ b/logic/models/edit.go @@ -164,12 +164,12 @@ func handleEdit(handle *Handle) { return c.Error500(err) } - c.Logger.Info("res", "id", lastLayer.Id, "start", lastLayer.Range_start, "end", lastLayer.Range_end) + c.Logger.Info("res", "id", lastLayer.Id, "start", lastLayer.Range_start, "end", lastLayer.Range_end, "shape", fmt.Sprintf("%d, 1", lastLayer.Range_end-lastLayer.Range_start + 1)) layers = append(layers, layerdef{ id: lastLayer.Id, LayerType: LAYER_DENSE, - Shape: fmt.Sprintf("%d, 1", lastLayer.Range_end-lastLayer.Range_start), + Shape: fmt.Sprintf("%d, 1", lastLayer.Range_end-lastLayer.Range_start + 1), }) /* diff --git a/logic/models/train/train.go b/logic/models/train/train.go index ef83c76..ecd9cb4 100644 --- a/logic/models/train/train.go +++ b/logic/models/train/train.go @@ -283,7 +283,7 @@ func trainDefinitionExp(c *Context, model *BaseModel, definition_id string, load got = append(got, layerrow{ LayerType: LAYER_DENSE, - Shape: fmt.Sprintf("%d", exp.end-exp.start), + Shape: fmt.Sprintf("%d", exp.end-exp.start + 1), ExpType: 2, LayerNum: i, })