sylv/fyp
1
0
Fork 0
Code Issues 29 Pull Requests Packages Projects 1 Releases Wiki Activity

Started working on moving to torch

Browse Source
This commit is contained in:
Andre Henriques
2024-04-19 15:39:51 +01:00
parent 2fa7680d0b
commit 28707b3f1b
28 changed files with 1082 additions and 430 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
logic/models/edit.go
View File

@@ -24,7 +24,7 @@ func handleEdit(handle *Handle) {
return c.Error500(err)
}
cls, err := model_classes.ListClasses(c, model.Id)
cls, err := model_classes.ListClassesJSON(c, model.Id)
if err != nil {
return c.Error500(err)
}
@@ -35,9 +35,9 @@ func handleEdit(handle *Handle) {
}
type ReturnType struct {
Classes []*model_classes.ModelClass `json:"classes"`
HasData bool `json:"has_data"`
NumberOfInvalidImages int `json:"number_of_invalid_images"`
Classes []*model_classes.ModelClassJSON `json:"classes"`
HasData bool `json:"has_data"`
NumberOfInvalidImages int `json:"number_of_invalid_images"`
}
c.ShowMessage = false
@@ -109,7 +109,7 @@ func handleEdit(handle *Handle) {
layers := []layerdef{}
for _, def := range defs {
if def.Status == MODEL_DEFINITION_STATUS_TRAINING {
if def.Status == DEFINITION_STATUS_TRAINING {
rows, err := c.Db.Query("select id, layer_type, shape from model_definition_layer where def_id=$1 order by layer_order asc;", def.Id)
if err != nil {
return c.Error500(err)
@@ -166,7 +166,7 @@ func handleEdit(handle *Handle) {
for i, def := range defs {
var lay *[]layerdef = nil
if def.Status == MODEL_DEFINITION_STATUS_TRAINING && !setLayers {
if def.Status == DEFINITION_STATUS_TRAINING && !setLayers {
lay = &layers
setLayers = true
}