fix: closes #63

2024-02-10 09:41:16 +00:00
parent bca44f9ba5
commit 3cd8acb919
2 changed files with 3 additions and 3 deletions
--- 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,
 	})