predict.yaml

compute:
  batch_size: 128
  epochs: 90
  print_freq: 10
  resume: null
  seed: null
  start_epoch: 0
  use_cuda: true
  workers: 8
dataset:
  class_names:
  - completely_round_smooth
  - in_between_smooth
  - cigar_shaped_smooth
  - edge_on
  - spiral
  clf_dir: classification
  clf_images_raw_dir: images_raw
  clf_images_test_dir: images_test
  clf_images_train_val_dir: images_train_val
  clf_labels_file: clf_labels.csv
  clf_labels_test_file: clf_labels_test.csv
  clf_labels_train_val_file: clf_labels_train_val.csv
  dir: dataset
  name: galaxy-zoo
  pred_dir: prediction
  predictions_file: predictions.csv
  raw_table: raw_table
  reg_dir: regression
  reg_images_test_dir: images_test_rev1
  reg_images_train_dir: images_training_rev1
  reg_labels_file: training_solutions_rev1.csv
  split_table: split_table
  test_split_ratio: 0.1
  val_split_ratio: 0.1
  version: latest
debug: false
distributed:
  dist_backend: nccl
  dist_url: tcp://224.66.41.62:23456
  gpu: null
  multiprocessing_distributed: false
  ngpus_per_node: 1
  rank: -1
  use: false
  world_size: -1
ensembling:
  n_estimators: 50
  use: false
exp:
  evaluate: true
  name: null
  task: classification
  test: true
model:
  arch: resnet18
  freeze: false
  output_constraints: true
  path: null
  pretrained: false
output: predictions/predictions.csv
template: all_ones_benchmark.csv