syrnt_v2 / config.json
johnlockejrr's picture
Trained for 10 epochs
f5e135a verified
{"n_classes": 3, "n_epochs": 6, "input_height": 448, "input_width": 224, "weight_decay": 1e-06, "n_batch": 1, "learning_rate": 0.0001, "patches": true, "augmentation": true, "flip_aug": false, "blur_aug": false, "padding_white": false, "padding_black": false, "scaling": true, "degrading": false, "brightening": false, "binarization": false, "adding_rgb_background": false, "adding_rgb_foreground": false, "add_red_textlines": false, "channels_shuffling": false, "dir_train": "/home/incognito/sbb_pixelwise_segmentation/dataset_syrnt/train", "dir_eval": "/home/incognito/sbb_pixelwise_segmentation/dataset_syrnt/eval", "dir_output": "/home/incognito/sbb_pixelwise_segmentation/syrnt_v2", "pretraining": true, "scaling_bluring": false, "scaling_binarization": false, "rotation": false, "rotation_not_90": false, "scaling_brightness": false, "scaling_flip": false, "thetha": [10, -10], "shuffle_indexes": null, "blur_k": ["blur", "guass", "median"], "scales": [0.6, 0.7, 0.8, 0.9, 1.1, 1.2, 1.4], "degrade_scales": [0.2, 0.4], "brightness": [1.3, 1.5, 1.7, 2], "flip_index": [0, 1, -1], "continue_training": true, "transformer_patchsize_x": null, "transformer_patchsize_y": null, "transformer_num_patches_xy": null, "transformer_projection_dim": 64, "transformer_mlp_head_units": [128, 64], "transformer_layers": 8, "transformer_num_heads": 4, "transformer_cnn_first": true, "index_start": 4, "dir_of_start_model": "/home/incognito/sbb_pixelwise_segmentation/syrnt_v2/model_3", "is_loss_soft_dice": false, "weighted_loss": false, "data_is_provided": true, "task": "segmentation", "f1_threshold_classification": null, "classification_classes_name": null, "backbone_type": "nontransformer", "dir_img_bin": null, "number_of_backgrounds_per_image": 1, "dir_rgb_backgrounds": null, "dir_rgb_foregrounds": null, "seed": 915004655}