global:
  data_path: /content/UCMerced_LandUse          # Path to folder containing samples
  number_of_bands: 3                            # Number of bands in input images
  model_name: inception                         # One of unet, unetsmall, checkpointed_unet, ternausnet, or inception
  bucket_name:                                  # name of the S3 bucket where data is stored. Leave blank if using local files
  classify: True                                # Set to True for classification tasks and False for semantic segmentation
  num_classes: 21
  sample_size: 256

models:
  inception:
    pretrained: False                           # using local checkpoints only

training:
  output_path: /content/output                  # Path to folder where files containing weights will be written
  batch_size: 30                                # Size of each batch
  num_epochs: 15                                # Number of epochs
  learning_rate: 0.001                          # Initial learning rate
  weight_decay: 0                               # Value for weight decay (each epoch)
  gamma: 0.1                                    # Multiple for learning rate decay
  step_size: 7                                  # Apply gamma every step_size
  class_weights: []                             # Weights to apply to each class. A value > 1.0 will apply more weights to the learning of the class.
  num_trn_samples: 1680
  num_val_samples: 420