Veith_et_al_2024/behavior/sleap_model/230817_101930.single_instance.n=121/training_config.json

{
    "data": {
        "labels": {
            "training_labels": "C:/Users/jlab/Documents/Ana/labelsF9.slp",
            "validation_labels": null,
            "validation_fraction": 0.1,
            "test_labels": null,
            "split_by_inds": false,
            "training_inds": [
                18,
                5,
                98,
                10,
                54,
                62,
                43,
                76,
                66,
                42,
                36,
                34,
                29,
                48,
                103,
                11,
                23,
                27,
                57,
                24,
                37,
                113,
                118,
                74,
                112,
                104,
                44,
                46,
                89,
                0,
                31,
                73,
                88,
                49,
                12,
                30,
                20,
                59,
                25,
                110,
                109,
                100,
                84,
                72,
                32,
                6,
                96,
                93,
                101,
                81,
                9,
                21,
                67,
                77,
                51,
                15,
                35,
                111,
                70,
                53,
                119,
                69,
                95,
                114,
                55,
                2,
                19,
                71,
                87,
                105,
                39,
                94,
                52,
                117,
                107,
                90,
                16,
                63,
                99,
                82,
                92,
                33,
                83,
                65,
                26,
                64,
                106,
                120,
                75,
                56,
                91,
                1,
                38,
                97,
                85,
                41,
                47,
                14,
                68,
                108,
                102,
                22,
                50,
                115,
                79,
                60,
                4,
                28,
                8
            ],
            "validation_inds": [
                86,
                78,
                3,
                45,
                58,
                13,
                17,
                116,
                80,
                40,
                7,
                61
            ],
            "test_inds": null,
            "search_path_hints": [
                "",
                ""
            ],
            "skeletons": [
                {
                    "directed": true,
                    "graph": {
                        "name": "Skeleton-1",
                        "num_edges_inserted": 8
                    },
                    "links": [
                        {
                            "edge_insert_idx": 0,
                            "key": 0,
                            "source": {
                                "py/object": "sleap.skeleton.Node",
                                "py/state": {
                                    "py/tuple": [
                                        "s1",
                                        1.0
                                    ]
                                }
                            },
                            "target": {
                                "py/object": "sleap.skeleton.Node",
                                "py/state": {
                                    "py/tuple": [
                                        "s2",
                                        1.0
                                    ]
                                }
                            },
                            "type": {
                                "py/reduce": [
                                    {
                                        "py/type": "sleap.skeleton.EdgeType"
                                    },
                                    {
                                        "py/tuple": [
                                            1
                                        ]
                                    }
                                ]
                            }
                        },
                        {
                            "edge_insert_idx": 1,
                            "key": 0,
                            "source": {
                                "py/id": 2
                            },
                            "target": {
                                "py/object": "sleap.skeleton.Node",
                                "py/state": {
                                    "py/tuple": [
                                        "s3",
                                        1.0
                                    ]
                                }
                            },
                            "type": {
                                "py/id": 3
                            }
                        },
                        {
                            "edge_insert_idx": 2,
                            "key": 0,
                            "source": {
                                "py/id": 4
                            },
                            "target": {
                                "py/object": "sleap.skeleton.Node",
                                "py/state": {
                                    "py/tuple": [
                                        "s4",
                                        1.0
                                    ]
                                }
                            },
                            "type": {
                                "py/id": 3
                            }
                        },
                        {
                            "edge_insert_idx": 3,
                            "key": 0,
                            "source": {
                                "py/id": 5
                            },
                            "target": {
                                "py/object": "sleap.skeleton.Node",
                                "py/state": {
                                    "py/tuple": [
                                        "s5",
                                        1.0
                                    ]
                                }
                            },
                            "type": {
                                "py/id": 3
                            }
                        },
                        {
                            "edge_insert_idx": 4,
                            "key": 0,
                            "source": {
                                "py/id": 6
                            },
                            "target": {
                                "py/object": "sleap.skeleton.Node",
                                "py/state": {
                                    "py/tuple": [
                                        "s6",
                                        1.0
                                    ]
                                }
                            },
                            "type": {
                                "py/id": 3
                            }
                        },
                        {
                            "edge_insert_idx": 5,
                            "key": 0,
                            "source": {
                                "py/id": 7
                            },
                            "target": {
                                "py/object": "sleap.skeleton.Node",
                                "py/state": {
                                    "py/tuple": [
                                        "s7",
                                        1.0
                                    ]
                                }
                            },
                            "type": {
                                "py/id": 3
                            }
                        },
                        {
                            "edge_insert_idx": 6,
                            "key": 0,
                            "source": {
                                "py/id": 8
                            },
                            "target": {
                                "py/object": "sleap.skeleton.Node",
                                "py/state": {
                                    "py/tuple": [
                                        "left_eye",
                                        1.0
                                    ]
                                }
                            },
                            "type": {
                                "py/id": 3
                            }
                        },
                        {
                            "edge_insert_idx": 7,
                            "key": 0,
                            "source": {
                                "py/id": 8
                            },
                            "target": {
                                "py/object": "sleap.skeleton.Node",
                                "py/state": {
                                    "py/tuple": [
                                        "right_eye",
                                        1.0
                                    ]
                                }
                            },
                            "type": {
                                "py/id": 3
                            }
                        },
                        {
                            "key": 0,
                            "source": {
                                "py/id": 9
                            },
                            "target": {
                                "py/id": 10
                            },
                            "type": {
                                "py/reduce": [
                                    {
                                        "py/type": "sleap.skeleton.EdgeType"
                                    },
                                    {
                                        "py/tuple": [
                                            2
                                        ]
                                    }
                                ]
                            }
                        },
                        {
                            "key": 0,
                            "source": {
                                "py/id": 10
                            },
                            "target": {
                                "py/id": 9
                            },
                            "type": {
                                "py/id": 11
                            }
                        }
                    ],
                    "multigraph": true,
                    "nodes": [
                        {
                            "id": {
                                "py/id": 1
                            }
                        },
                        {
                            "id": {
                                "py/id": 2
                            }
                        },
                        {
                            "id": {
                                "py/id": 4
                            }
                        },
                        {
                            "id": {
                                "py/id": 5
                            }
                        },
                        {
                            "id": {
                                "py/id": 6
                            }
                        },
                        {
                            "id": {
                                "py/id": 7
                            }
                        },
                        {
                            "id": {
                                "py/id": 8
                            }
                        },
                        {
                            "id": {
                                "py/id": 9
                            }
                        },
                        {
                            "id": {
                                "py/id": 10
                            }
                        }
                    ]
                }
            ]
        },
        "preprocessing": {
            "ensure_rgb": false,
            "ensure_grayscale": false,
            "imagenet_mode": null,
            "input_scaling": 1.0,
            "pad_to_stride": 16,
            "resize_and_pad_to_target": true,
            "target_height": 336,
            "target_width": 336
        },
        "instance_cropping": {
            "center_on_part": null,
            "crop_size": null,
            "crop_size_detection_padding": 16
        }
    },
    "model": {
        "backbone": {
            "leap": null,
            "unet": {
                "stem_stride": null,
                "max_stride": 16,
                "output_stride": 2,
                "filters": 64,
                "filters_rate": 2.0,
                "middle_block": true,
                "up_interpolate": false,
                "stacks": 1
            },
            "hourglass": null,
            "resnet": null,
            "pretrained_encoder": null
        },
        "heads": {
            "single_instance": {
                "part_names": [
                    "s1",
                    "s2",
                    "s3",
                    "s4",
                    "s5",
                    "s6",
                    "s7",
                    "left_eye",
                    "right_eye"
                ],
                "sigma": 1.0,
                "output_stride": 2,
                "loss_weight": 1.0,
                "offset_refinement": false
            },
            "centroid": null,
            "centered_instance": null,
            "multi_instance": null,
            "multi_class_bottomup": null,
            "multi_class_topdown": null
        },
        "base_checkpoint": null
    },
    "optimization": {
        "preload_data": true,
        "augmentation_config": {
            "rotate": true,
            "rotation_min_angle": -15.0,
            "rotation_max_angle": 15.0,
            "translate": false,
            "translate_min": -5,
            "translate_max": 5,
            "scale": true,
            "scale_min": 0.9,
            "scale_max": 1.1,
            "uniform_noise": false,
            "uniform_noise_min_val": 0.0,
            "uniform_noise_max_val": 10.0,
            "gaussian_noise": false,
            "gaussian_noise_mean": 5.0,
            "gaussian_noise_stddev": 1.0,
            "contrast": true,
            "contrast_min_gamma": 0.8,
            "contrast_max_gamma": 1.2,
            "brightness": true,
            "brightness_min_val": 0.0,
            "brightness_max_val": 5.0,
            "random_crop": false,
            "random_crop_height": 256,
            "random_crop_width": 256,
            "random_flip": false,
            "flip_horizontal": true
        },
        "online_shuffling": true,
        "shuffle_buffer_size": 128,
        "prefetch": true,
        "batch_size": 4,
        "batches_per_epoch": 200,
        "min_batches_per_epoch": 200,
        "val_batches_per_epoch": 10,
        "min_val_batches_per_epoch": 10,
        "epochs": 200,
        "optimizer": "adam",
        "initial_learning_rate": 0.0001,
        "learning_rate_schedule": {
            "reduce_on_plateau": true,
            "reduction_factor": 0.5,
            "plateau_min_delta": 1e-06,
            "plateau_patience": 5,
            "plateau_cooldown": 3,
            "min_learning_rate": 1e-08
        },
        "hard_keypoint_mining": {
            "online_mining": false,
            "hard_to_easy_ratio": 2.0,
            "min_hard_keypoints": 2,
            "max_hard_keypoints": null,
            "loss_scale": 5.0
        },
        "early_stopping": {
            "stop_training_on_plateau": true,
            "plateau_min_delta": 1e-08,
            "plateau_patience": 10
        }
    },
    "outputs": {
        "save_outputs": true,
        "run_name": "230817_101930.single_instance.n=121",
        "run_name_prefix": "",
        "run_name_suffix": "",
        "runs_folder": "C:/Users/jlab/Documents/Ana\\models",
        "tags": [
            ""
        ],
        "save_visualizations": true,
        "delete_viz_images": true,
        "zip_outputs": false,
        "log_to_csv": true,
        "checkpointing": {
            "initial_model": false,
            "best_model": true,
            "every_epoch": false,
            "latest_model": false,
            "final_model": false
        },
        "tensorboard": {
            "write_logs": false,
            "loss_frequency": "epoch",
            "architecture_graph": false,
            "profile_graph": false,
            "visualizations": true
        },
        "zmq": {
            "subscribe_to_controller": true,
            "controller_address": "tcp://127.0.0.1:9000",
            "controller_polling_timeout": 10,
            "publish_updates": true,
            "publish_address": "tcp://127.0.0.1:9001"
        }
    },
    "name": "",
    "description": "",
    "sleap_version": "1.3.1",
    "filename": "C:/Users/jlab/Documents/Ana\\models\\230817_101930.single_instance.n=121\\training_config.json"
}