From d77f98298108c65a30727580e96e28be361baa0f Mon Sep 17 00:00:00 2001 From: Ethan White Date: Fri, 12 Jul 2024 13:27:48 -0400 Subject: [PATCH] Remove deepforest config to confirm the fix is the runner type --- deepforest_config.yml | 60 ------------------------------------------- 1 file changed, 60 deletions(-) delete mode 100644 deepforest_config.yml diff --git a/deepforest_config.yml b/deepforest_config.yml deleted file mode 100644 index c4b17e7..0000000 --- a/deepforest_config.yml +++ /dev/null @@ -1,60 +0,0 @@ -# Config file for DeepForest pytorch module - -# Cpu workers for data loaders -# Dataloaders -workers: 1 -devices: auto -accelerator: cpu -batch_size: 1 - -# Model Architecture -architecture: 'retinanet' -num_classes: 1 -nms_thresh: 0.05 - -# Architecture specific params -retinanet: - # Non-max supression of overlapping predictions - score_thresh: 0.1 - -train: - csv_file: - root_dir: - - # Optimizer initial learning rate - lr: 0.001 - scheduler: - type: - params: - # Common parameters - T_max: 10 - eta_min: 0.00001 - lr_lambda: "lambda epoch: 0.95 ** epoch" # For lambdaLR and multiplicativeLR - step_size: 30 # For stepLR - gamma: 0.1 # For stepLR, multistepLR, and exponentialLR - milestones: [50, 100] # For multistepLR - - # ReduceLROnPlateau parameters (used if type is not explicitly mentioned) - mode: "min" - factor: 0.1 - patience: 10 - threshold: 0.0001 - threshold_mode: "rel" - cooldown: 0 - min_lr: 0 - eps: 1e-08 - - # Print loss every n epochs - epochs: 1 - # Useful debugging flag in pytorch lightning, set to True to get a single batch of training to test settings. - fast_dev_run: False - # pin images to GPU memory for fast training. This depends on GPU size and number of images. - preload_images: False - -validation: - # callback args - csv_file: - root_dir: - # Intersection over union evaluation - iou_threshold: 0.4 - val_accuracy_interval: 20