FaceOcc / training_config.json
mertakin's picture
Add FaceOcc v1.0.0 model release
056d92f verified
Raw
History Blame Contribute Delete
563 Bytes
{
"objective": "OHEM-BCE",
"ohem_mode": "global",
"ohem_threshold_probability": 0.7,
"ohem_minimum_total_hard_pixels": 65535,
"epochs": 30,
"batch_size": 16,
"seed": 42,
"precision": "fp32",
"normalization": "imagenet",
"checkpoint_selection": "maximum COFW validation face IoU at p=0.5",
"best_epoch": 24,
"best_cofw_iou": 0.936523,
"train_size": 27792,
"validation_size": 500,
"device": "NVIDIA GeForce RTX 3060 12 GB",
"torch_version": "2.13.0+cu132",
"cuda_version": "13.2",
"python_version": "3.12.13"
}