Fork: 0
diego / RARP
Transfer to URL with SHA Find file
Newer
Older
RARP / eval-E1.yaml
@delAguila delAguila on 22 Nov 2024 2 KB init comit
Raw Blame History 
models:
  ResNet50_ckpt:
    - ./log_ResNet50_X6/lightning_logs/version_5/checkpoints/epoch=25-step=104.ckpt
    - ./log_ResNet50_X6/lightning_logs/version_6/checkpoints/epoch=29-step=90.ckpt
    - ./log_ResNet50_X6/lightning_logs/version_7/checkpoints/epoch=14-step=45.ckpt
    - ./log_ResNet50_X6/lightning_logs/version_8/checkpoints/epoch=20-step=84.ckpt
    - ./log_ResNet50_X6/lightning_logs/version_9/checkpoints/epoch=44-step=180.ckpt
  ResNet18_ckpt:
    #- ./log_restnet18_X6/lightning_logs/version_14/checkpoints/epoch=22-step=92.ckpt
  MovilNetV2_ckpt:
    #- ./log_MobileNetV2_X6/lightning_logs/version_14/checkpoints/epoch=41-step=168.ckpt
  EfficientNetV2_ckpt:
    - ./log_EfficientNetV2_X6/lightning_logs/version_5/checkpoints/epoch=25-step=104.ckpt
    - ./log_EfficientNetV2_X6/lightning_logs/version_6/checkpoints/epoch=23-step=72.ckpt
    - ./log_EfficientNetV2_X6/lightning_logs/version_7/checkpoints/epoch=49-step=150.ckpt
    - ./log_EfficientNetV2_X6/lightning_logs/version_8/checkpoints/epoch=23-step=96.ckpt
    - ./log_EfficientNetV2_X6/lightning_logs/version_9/checkpoints/epoch=19-step=80.ckpt
    

dataset_type: YOLO_ROI #full_size, Manual_ROI, Manual_ROIwD, YOLO_ROI

YOLO_Accuracy_min_ROI: 0.40

Fold_Num: 4

Fold_Ensamble: False

# este es el mejor hasta ahora
#- ./log_ResNet50_X6/lightning_logs/version_14/checkpoints/epoch=23-step=96.ckpt
#- ./log_ResNet50_X6/lightning_logs/version_9/checkpoints/epoch=44-step=180.ckpt

#- ./log_EfficientNetV2_X6/lightning_logs/version_8/checkpoints/epoch=23-step=96.ckpt
#- ./log_EfficientNetV2_X5/lightning_logs/version_10/checkpoints/epoch=25-step=104.ckpt

#- ./log_ResNet50_X6/lightning_logs/version_6/checkpoints/epoch=29-step=90.ckpt
#- ./log_ResNet50_X6/lightning_logs/version_14/checkpoints/epoch=23-step=96.ckpt

#- ./log_ResNet50_X6/lightning_logs/version_10/checkpoints/epoch=26-step=108.ckpt
    #- ./log_ResNet50_X6/lightning_logs/version_11/checkpoints/epoch=23-step=72.ckpt
    #- ./log_ResNet50_X6/lightning_logs/version_12/checkpoints/epoch=22-step=69.ckpt
    #- ./log_ResNet50_X6/lightning_logs/version_13/checkpoints/epoch=10-step=44.ckpt
    #- ./log_ResNet50_X6/lightning_logs/version_14/checkpoints/epoch=23-step=96.ckpt