HybVelEst_wLoss_ModFc_v2/config.json

{
    "DEVICE": "cuda",
    "FLOAT32_MATMUL_PRECISION": "high",
    "DIM_SIZE_FACTOR": 1,
    "IMAGE_HEIGHTS_MEDIAN": 224,
    "IMAGE_WIDTHS_MEDIAN": 224,
    "IMAGE_CHANNELS": 3,
    "IMAGE_MEAN": [
        0.45484897,
        0.47033188,
        0.47973604
    ],
    "IMAGE_STD": [
        0.24885239,
        0.24789241,
        0.24780511
    ],
    "N_SAMPLES": 5202,
    "BATCH_SIZE": 128,
    "NUM_WORKERS": 20,
    "TRAIN_PROPORTION": 0.8,
    "VAL_PROPORTION": 0.8,
    "N_EPOCHS": 80,
    "LEARNING_RATE": 0.0001,
    "SEED": 42,
    "EARLY_STOPPING_PATIENCE": 10,
    "TRAINER_ACCELERATOR": "gpu",
    "TRAINER_PRECISION": "bf16-mixed",
    "IO_DATASET_MAP_LOCAL_PATH": "model/data/CSV/dataset_v3/tabular_features_selected.csv",
    "INPUT_IMAGES_CSV_INDEX": "image_relpath",
    "OUTPUT_LABEL_CSV_INDEX": "targetVariable",
    "ENCODER_IMAGEVELOCITYESTIMATOR_SERIALIZED_PATH": "model/experiments/ImagVelEst_aug_ModFc_EmbHead_l1Loss_v1/serialized/ImagVelEst_aug_ModFc_EmbHead_l1Loss_v1_weights.pth",
    "MODEL_PREFIX": "HybVelEst_wLoss_ModFc_v2",
    "EXPERIMENTS_ROOT_DIR": "model/experiments",
    "INFERENCE_MODE": false,
    "METRICS_MODE": false
}