random_seed: 42 model_config { num_layers: 18 model_input_width: 960 model_input_height: 544 model_input_channels: 3 all_projections: true arch: "vanilla_unet_dynamic" use_batch_norm: true training_precision { backend_floatx: FLOAT32 } } training_config { batch_size: 3 epochs: 100 log_summary_steps: 10 checkpoint_interval: 1 loss: "cross_entropy" learning_rate:0.00001 regularizer { type: L2 weight: 0 } optimizer { adam { epsilon: 9.99999993923e-09 beta1: 0.899999976158 beta2: 0.999000012875 } } } dataset_config { dataset: "custom" augment: False input_image_type: "grayscale" train_images_path:"/workspace/tao-experiments/data/itri_0316/images/train" train_masks_path:"/workspace/tao-experiments/data/itri_0316/masks/train" val_images_path:"/workspace/tao-experiments/data/itri_0316/images/val" val_masks_path:"/workspace/tao-experiments/data/itri_0316/masks/val" test_images_path:"/workspace/tao-experiments/data/itri_0316/images/test" data_class_config { target_classes { name: "OK" mapping_class: "background" label_id: 0 } target_classes { name: "NG" mapping_class: "person" label_id: 1 } } }