robossembler/webstudio
7
0
Fork 0
Code Issues 8 Pull requests Projects Releases Packages Wiki Activity Actions

progress

Browse source
This commit is contained in:
IDONTSUDO 2024-04-26 15:51:20 +03:00
parent c4ddb3dc8c
commit f0ed478b36
2 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
server/src/features/weights/models/weights_database_model.ts
View file

@ -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;

8
server/src/features/weights/models/weights_validation_model.ts
View file

@ -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:{
type:Number,
epoch: {
type: Number,
},
datasetId: {
type: Schema.Types.ObjectId,