package cn.iocoder.yudao.module.llm.controller.admin.training.vo;

import lombok.*;
import java.util.*;
import io.swagger.v3.oas.annotations.media.Schema;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
import java.math.BigDecimal;
import org.springframework.format.annotation.DateTimeFormat;
import java.time.LocalDateTime;

import static cn.iocoder.yudao.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND;

@Schema(description = "管理后台 - 训练分页 Request VO")
@Data
@EqualsAndHashCode(callSuper = true)
@ToString(callSuper = true)
public class TrainingPageReqVO extends PageParam {

    @Schema(description = "基座模型名称", example = "张三")
    private String modelName;

    @Schema(description = "介绍", example = "你说的对")
    private String description;

    @Schema(description = "训练轮次")
    private Integer trainingEpochs;

    @Schema(description = "启用Lora")
    private Boolean useLora;

    @Schema(description = "训练批次")
    private Integer trainingBatchSize;

    @Schema(description = "评估批次")
    private Integer evaluationBatchSize;

    @Schema(description = "梯度累加步数")
    private Integer gradientAccumulationSteps;

    @Schema(description = "使用GPU数量", example = "22492")
    private Integer gpuCount;

    @Schema(description = "学习率")
    private BigDecimal learningRate;

    @Schema(description = "最大长度")
    private Integer maxLength;

    @Schema(description = "记录创建时间")
    private LocalDateTime createdAt;

    @Schema(description = "记录更新时间")
    private LocalDateTime updatedAt;

}