random_seed: 42 lpr_config { hidden_units: 512 max_label_length: 20 arch: "baseline" nlayers: 18 #setting nlayers to be 10 to use baseline10 model } training_config { batch_size_per_gpu: 16 num_epochs: 100 learning_rate { soft_start_annealing_schedule { min_learning_rate: 1e-6 max_learning_rate: 1e-4 soft_start: 0.001 annealing: 0.5 } } regularizer { type: L2 weight: 5e-4 } } eval_config { validation_period_during_training: 5 batch_size: 1 } augmentation_config { output_width: 96 output_height: 48 output_channel: 3 max_rotate_degree: 45 rotate_prob: 0.5 gaussian_kernel_size: 5 gaussian_kernel_size: 7 gaussian_kernel_size: 15 blur_prob: 0.4 reverse_color_prob: 0.5 keep_original_prob: 0.3 } dataset_config { data_sources: { label_directory_path: "/workspace/tao-experiments/lprnet/data/LPR_Final_data/train/labels" image_directory_path: "/workspace/tao-experiments/lprnet/data/LPR_Final_data/train/images" } characters_list_file: "/workspace/tao-experiments/lprnet/specs/us_lp_characters.txt" validation_data_sources: { label_directory_path: "/workspace/tao-experiments/lprnet/data/LPR_Final_data/val/labels" image_directory_path: "/workspace/tao-experiments/lprnet/data/LPR_Final_data/val/images" } }