random_seed: 42 model_config { model_input_width: 1280 model_input_height: 704 model_input_channels: 3 num_layers: 16 all_projections: true arch: "vgg" use_batch_norm: False training_precision { backend_floatx: FLOAT32 } } training_config { batch_size: 4 epochs: 1100 log_summary_steps: 10 checkpoint_interval: 100 loss: "cross_entropy" learning_rate:0.0001 regularizer { type: L2 weight: 2e-6 } optimizer { adam { epsilon: 9.99999993923e-09 beta1: 0.899999976158 beta2: 0.999000012875 } } } dataset_config { dataset: "custom" augment: false input_image_type: "color" train_images_path:"/workspace/tao-experiments/data/images/train" train_masks_path:"/workspace/tao-experiments/data/masks/train" val_images_path:"/workspace/tao-experiments/data/images/val" val_masks_path:"/workspace/tao-experiments/data/masks/val" test_images_path:"/workspace/tao-experiments/data/images/test" data_class_config { target_classes { name: "Background" mapping_class: "Background" label_id: 0 } target_classes { name: "Joint" mapping_class: "Joint" label_id: 1 } target_classes { name: "Robot" mapping_class: "Robot" label_id: 2 } target_classes { name: "End" mapping_class: "End" label_id: 3 } target_classes { name: "Stem" mapping_class: "Stem" label_id: 4 } target_classes { name: "Leaf" mapping_class: "Leaf" label_id: 5 } } }