progress

server/src/features/weights/models/weights_database_model.ts

@@ -2,6 +2,8 @@ import { IsNumber, IsString } from "class-validator";
 import { IWeightModel } from "./weights_validation_model";
 
 export class WeightValidationModel implements IWeightModel {
+  @IsNumber()
+  public numberOfTrainedEpochs: number;
   public processStatus: string;
   @IsNumber()
   public epoch: number;

server/src/features/weights/models/weights_validation_model.ts

@@ -9,6 +9,7 @@ export interface IWeightModel {
   project: string | IProjectModel;
   processStatus: string;
   epoch: number;
+  numberOfTrainedEpochs: number;
 }
 export const WeightSchema = new Schema({
   name: {
@@ -17,6 +18,9 @@ export const WeightSchema = new Schema({
   local_path: {
     type: String,
   },
+  numberOfTrainedEpochs: {
+    type: Number,
+  },
   neuralNetworkName: {
     type: String,
   },
@@ -30,8 +34,8 @@ export const WeightSchema = new Schema({
     type: Boolean,
     default: false,
   },
-  epoch:{
+  epoch: {
-    type:Number,
+    type: Number,
   },
   datasetId: {
     type: Schema.Types.ObjectId,