diff --git a/config/experiments/demo_med/KA_130923_fbFMFDe1_lung_description_decision.yaml b/config/experiments/demo_med/KA_130923_fbFMFDe1_lung_description_decision.yaml new file mode 100644 index 00000000..871cd1f3 --- /dev/null +++ b/config/experiments/demo_med/KA_130923_fbFMFDe1_lung_description_decision.yaml @@ -0,0 +1,8 @@ +# @package _global_ +defaults: + - override /models: classification/lung_description_decision.yaml + - override /datasets: classification/lung_description_decision_datamodule.yaml + - override /callbacks: lung_description_decision + +project: "lung_description_decision" +task: "table-classification" diff --git a/config/experiments/demo_med/KG_100323_ianfq90_lion_test.yaml b/config/experiments/demo_med/KG_100323_ianfq90_lion_test.yaml new file mode 100644 index 00000000..2ff8ee0b --- /dev/null +++ b/config/experiments/demo_med/KG_100323_ianfq90_lion_test.yaml @@ -0,0 +1,19 @@ +# @package _global_ +defaults: + - override /models: semantic-segmentation/unet + - override /datasets: semantic-segmentation/segmentation_brain.yaml + - override /optimizers: lion.yaml + - override /augmentations_train: none + - override /augmentations_val: none + - override /augmentations_test: none + - override /losses: segmentation_losses.yaml + +models: + in_channels: 1 +project: "segmentation" +task: "image-segmentation" +random_seed: 42 +stop_param: 10 +epochs: 10 +weights_freq: 1 +batch_size: 4 \ No newline at end of file diff --git a/config/experiments/demo_med/KG_230323_sjdv8f20_segformer.yaml b/config/experiments/demo_med/KG_230323_sjdv8f20_segformer.yaml new file mode 100644 index 00000000..f9cba032 --- /dev/null +++ b/config/experiments/demo_med/KG_230323_sjdv8f20_segformer.yaml @@ -0,0 +1,26 @@ +# @package _global_ +defaults: + - override /models: semantic-segmentation/segformer_for_segmentation.yaml + - override /datasets: semantic-segmentation/segmentation_brain.yaml + - override /augmentations_train: none + - override /optimizers: adam.yaml + - override /augmentations_val: none + - override /augmentations_test: none + - override /losses: segmentation_losses.yaml + - override /callbacks: segmentation + + +models: + num_channels: 1 + num_labels: 1 + retain_dim: True +project: "segmentation" +task: "image-segmentation" +random_seed: 42 +stop_param: 3 +epochs: 3 +weights_freq: 1 +batch_size: 8 +accelerator: cpu # gpu +#override hydra/job_logging: stdout + diff --git a/config/experiments/demo_med/SK_100923_unet_brain_mri.yaml b/config/experiments/demo_med/SK_100923_unet_brain_mri.yaml new file mode 100644 index 00000000..73891dda --- /dev/null +++ b/config/experiments/demo_med/SK_100923_unet_brain_mri.yaml @@ -0,0 +1,17 @@ +# @package _global_ +defaults: + - override /models: semantic-segmentation/unet + - override /datasets: semantic-segmentation/segmentation_brain.yaml + - override /augmentations_train: none #segmentation_stroke.yaml + - override /augmentations_val: none + - override /augmentations_test: none + - override /losses: segmentation_losses.yaml +models: + in_channels: 1 +project: "segmentation" +task: "image-segmentation-complexing" +random_seed: 42 +stop_param: 10 +epochs: 300 +weights_freq: 1 +batch_size: 10 diff --git a/config/experiments/demo_med/SK_180822_qmciwj41_unet_brain.yaml b/config/experiments/demo_med/SK_180822_qmciwj41_unet_brain.yaml new file mode 100644 index 00000000..dd97d9f0 --- /dev/null +++ b/config/experiments/demo_med/SK_180822_qmciwj41_unet_brain.yaml @@ -0,0 +1,17 @@ +# @package _global_ +defaults: + - override /models: semantic-segmentation/unet + - override /datasets: semantic-segmentation/segmentation_brain.yaml + - override /augmentations_train: none #segmentation_stroke.yaml + - override /augmentations_val: none + - override /augmentations_test: none + - override /losses: segmentation_losses.yaml +models: + in_channels: 1 +project: "segmentation" +task: "image-segmentation" +random_seed: 42 +stop_param: 10 +epochs: 300 +weights_freq: 1 +batch_size: 10 diff --git a/config/experiments/demo_med/SK_180822_qwefnew31_yolov5_lungs.yaml b/config/experiments/demo_med/SK_180822_qwefnew31_yolov5_lungs.yaml new file mode 100644 index 00000000..17ac2ea2 --- /dev/null +++ b/config/experiments/demo_med/SK_180822_qwefnew31_yolov5_lungs.yaml @@ -0,0 +1,12 @@ +# @package _global_ +defaults: + - override /models: detection/yolov5 + - override /datasets: detection/detection_lungs + + +project: "lungs" +task: "image-detection" +random_seed: 43 +epochs: 300 +batch_size: 2 +weights_freq: 1 diff --git a/config/experiments/demo_med/SK_180822_qwefnew31_yolov5_lungs_seg.yaml b/config/experiments/demo_med/SK_180822_qwefnew31_yolov5_lungs_seg.yaml new file mode 100644 index 00000000..a22b1bea --- /dev/null +++ b/config/experiments/demo_med/SK_180822_qwefnew31_yolov5_lungs_seg.yaml @@ -0,0 +1,12 @@ +# @package _global_ +defaults: + - override /models: detection/yolov5_segmentation + - override /datasets: detection/detection_lungs + + +project: "lungs" +task: "image-detection" +random_seed: 43 +epochs: 20 +batch_size: 2 +weights_freq: 1 diff --git a/config/experiments/demo_med/SK_210722_vfdkmls3_deeplabv3.yaml b/config/experiments/demo_med/SK_210722_vfdkmls3_deeplabv3.yaml new file mode 100644 index 00000000..b7dae60b --- /dev/null +++ b/config/experiments/demo_med/SK_210722_vfdkmls3_deeplabv3.yaml @@ -0,0 +1,16 @@ +# @package _global_ +defaults: + - override /models: semantic-segmentation/deeplabv3plus.yaml + - override /datasets: semantic-segmentation/segmentation_arable.yaml + - override /optimizers: adam.yaml + - override /losses: segmentation_losses + - override /augmentations_train: none #segmentation_inference + - override /augmentations_val: none + - override /augmentations_test: none + - override /initializations: he + +project: "segmentation" +task: "image-segmentation" +random_seed: 42 +weights_freq: 1 +batch_size: 10 diff --git a/config/experiments/demo_med/SK_260722_frwmfrw4_faster_rcnn.yaml b/config/experiments/demo_med/SK_260722_frwmfrw4_faster_rcnn.yaml new file mode 100644 index 00000000..70980a4a --- /dev/null +++ b/config/experiments/demo_med/SK_260722_frwmfrw4_faster_rcnn.yaml @@ -0,0 +1,13 @@ +# @package _global_ +defaults: + - override /models: detection/faster_rcnn + - override /datasets: detection/detection_wheat + - override /optimizers: adam.yaml + + +project: "wheat" +task: "image-detection" +random_seed: 42 +epochs: 300 +batch_size: 2 +weights_freq: 1 diff --git a/config/experiments/demo_med/example_image_segmentation_unet.yaml b/config/experiments/demo_med/example_image_segmentation_unet.yaml index 41c45138..f48224ea 100644 --- a/config/experiments/demo_med/example_image_segmentation_unet.yaml +++ b/config/experiments/demo_med/example_image_segmentation_unet.yaml @@ -6,13 +6,13 @@ defaults: - override /augmentations_val: none - override /augmentations_test: none - override /losses: segmentation_losses.yaml + - override /clear_ml: enabled models: in_channels: 1 project: "segmentation" task: "image-segmentation" random_seed: 42 -stop_param: 10 epochs: 35 weights_freq: 1 batch_size: 10 -ckpt_path: https://api.blackhole.ai.innopolis.university/pretrained/segmentation_unet_brain.pt \ No newline at end of file +ckpt_path: https://api.blackhole.ai.innopolis.university/pretrained/med/stroke/baseline/unet_fscore94.pt diff --git a/config/experiments/demo_med/uwmadison/KG_150423_lsjdvnd8_segformer_uwmadison.yaml b/config/experiments/demo_med/uwmadison/KG_150423_lsjdvnd8_segformer_uwmadison.yaml new file mode 100644 index 00000000..3f0f0d93 --- /dev/null +++ b/config/experiments/demo_med/uwmadison/KG_150423_lsjdvnd8_segformer_uwmadison.yaml @@ -0,0 +1,37 @@ +# @package _global_ +defaults: + - override /models: semantic-segmentation/segformer_for_segmentation.yaml + - override /datasets: semantic-segmentation/segmentation_uwmadison + - override /optimizers: lion.yaml + - override /augmentations_train: none + - override /augmentations_val: none + - override /augmentations_test: none + - override /callbacks: multiclass_segmentation + - override /losses: multiclass_segmentation_losses + - override /loggers: wandb + +models: + num_channels: 1 + num_labels: 4 + retain_dim: True + +optimizers: + lr: 0.0003 + +project: "organs segmentation" +task: "multiclass-image-segmentation" +random_seed: 42 +weights_freq: 1 +batch_size: 16 +n_classes: 4 +epochs: 100 +accelerator: gpu +override hydra/job_logging: stdout + +wandb: + enable: True + project: uw_madison + entity: "k-galliamov" + group: none + job_type: training + diff --git a/config/experiments/demo_med/uwmadison/KG_260423_boqe16_unet_uwmadison.yaml b/config/experiments/demo_med/uwmadison/KG_260423_boqe16_unet_uwmadison.yaml new file mode 100644 index 00000000..bc34eb78 --- /dev/null +++ b/config/experiments/demo_med/uwmadison/KG_260423_boqe16_unet_uwmadison.yaml @@ -0,0 +1,38 @@ +# @package _global_ +defaults: + - override /models: semantic-segmentation/unet + - override /datasets: semantic-segmentation/segmentation_uwmadison + - override /optimizers: adam.yaml + - override /augmentations_train: none + - override /augmentations_val: none + - override /augmentations_test: none + - override /callbacks: multiclass_segmentation + - override /losses: multiclass_segmentation_losses + - override /loggers: wandb + +n_classes: 4 + +models: + in_channels: 1 + classes: ${n_classes} + encoder_name: resnet34 + +optimizers: + lr: 0.00477 + +project: "organs segmentation" +task: "multiclass-image-segmentation" +random_seed: 42 +weights_freq: 1 +batch_size: 8 +epochs: 25 +accelerator: gpu +devices: 1 +override hydra/job_logging: stdout + +wandb: + enable: True + project: uw_madison + entity: "k-galliamov" + job_type: training + diff --git a/config/experiments/demo_med/uwmadison/KG_260423_pgn135_deeplab_uwmadison.yaml b/config/experiments/demo_med/uwmadison/KG_260423_pgn135_deeplab_uwmadison.yaml new file mode 100644 index 00000000..a96a504b --- /dev/null +++ b/config/experiments/demo_med/uwmadison/KG_260423_pgn135_deeplab_uwmadison.yaml @@ -0,0 +1,41 @@ +# @package _global_ +defaults: + - override /models: semantic-segmentation/deeplabv3plus.yaml + - override /datasets: semantic-segmentation/segmentation_uwmadison + - override /optimizers: adam.yaml + - override /augmentations_train: ../augmentations/position/random_horizontal_flip.yaml + - override /augmentations_val: none + - override /augmentations_test: none + - override /callbacks: multiclass_segmentation + - override /losses: multiclass_segmentation_losses + - override /loggers: wandb + +models: + in_channels: 1 + classes: 4 + encoder_name: resnet34 + +optimizers: + lr: 0.004313519324382217 + +n_classes: 4 +project: "organs segmentation" +task: "multiclass-image-segmentation" +random_seed: 42 +weights_freq: 1 +ckpt_path: https://api.blackhole.ai.innopolis.university/pretrained/segmentation/uw_madison_deeplabv3plus.pickle + +# When tuning from the checkpoint, the number of epochs has to be set to at least +# the № of epoch on which the checkpoint was created + 1 (in this case, 39 + 1 = 40 would be a lower bound) +epochs: 50 +batch_size: 128 +accelerator: gpu +devices: 1 +override hydra/job_logging: stdout + +wandb: + enable: True + project: uw_madison + entity: "k-galliamov" + job_type: training + diff --git a/config/experiments/demo_nm/GR_050822_AGgrv2Dg_catboost_regressor_qm9.yaml b/config/experiments/demo_nm/GR_050822_AGgrv2Dg_catboost_regressor_qm9.yaml new file mode 100644 index 00000000..80d8431b --- /dev/null +++ b/config/experiments/demo_nm/GR_050822_AGgrv2Dg_catboost_regressor_qm9.yaml @@ -0,0 +1,10 @@ +# @package _global_ +defaults: + - override /models: regression/catboost_regression + - override /datasets: qm9 + - override /callbacks: regression + + +project: "qm9" +task: "qsar-regression" +random_seed: 42 diff --git a/config/experiments/demo_nm/GR_050822_AGgrv2Dg_catboost_regressor_qm9_non_pretrained.yaml b/config/experiments/demo_nm/GR_050822_AGgrv2Dg_catboost_regressor_qm9_non_pretrained.yaml new file mode 100644 index 00000000..f57ce619 --- /dev/null +++ b/config/experiments/demo_nm/GR_050822_AGgrv2Dg_catboost_regressor_qm9_non_pretrained.yaml @@ -0,0 +1,18 @@ +# @package _global_ +defaults: + - override /initializations: random + - override /models: regression/catboost_regression + - override /datasets: qm9_2 + - override /callbacks: regression + + + +project: "qm9" +task: "qsar-regression" +random_seed: 42 + +# ckpt_path: https://api.blackhole.ai.innopolis.university/pretrained/catboost_regression_qm9.pickle + + +models: + learning_rate: 0.1 diff --git a/config/experiments/demo_nm/GR_050822_AGgrv2Dg_catboost_regressor_qm9_pretrained.yaml b/config/experiments/demo_nm/GR_050822_AGgrv2Dg_catboost_regressor_qm9_pretrained.yaml new file mode 100644 index 00000000..8d2b7857 --- /dev/null +++ b/config/experiments/demo_nm/GR_050822_AGgrv2Dg_catboost_regressor_qm9_pretrained.yaml @@ -0,0 +1,13 @@ +# @package _global_ +defaults: + - initializations: + - override /models: regression/catboost_regression + - override /datasets: qm9_2 + - override /callbacks: regression + + +project: "qm9" +task: "qsar-regression" +random_seed: 42 + +ckpt_path: https://api.blackhole.ai.innopolis.university/pretrained/catboost_regression_qm9.pickle diff --git a/config/experiments/demo_nm/GR_061022_DkOLIn2s_vae_qm9_reverse.yaml b/config/experiments/demo_nm/GR_061022_DkOLIn2s_vae_qm9_reverse.yaml new file mode 100644 index 00000000..2fe0be67 --- /dev/null +++ b/config/experiments/demo_nm/GR_061022_DkOLIn2s_vae_qm9_reverse.yaml @@ -0,0 +1,18 @@ +# @package _global_ +defaults: + - override /models: text-vae/hier_vae.py + - override /datasets: qm9_selfies + - override /losses: vae + + +experiment_name: "GR_061022_DkOLIn2s_vae_qm9_reverse" +project: chem-vae +task: text-vae-reverse +random_seed: 42 +accelerator: cpu +devices: 1 +batch_size: 128 +num_workers: 0 + +datasets: + work_mode: reverse \ No newline at end of file diff --git a/config/experiments/demo_nm/GR_061022_skOLI42s_vae_qm9_forward.yaml b/config/experiments/demo_nm/GR_061022_skOLI42s_vae_qm9_forward.yaml new file mode 100644 index 00000000..a1e75125 --- /dev/null +++ b/config/experiments/demo_nm/GR_061022_skOLI42s_vae_qm9_forward.yaml @@ -0,0 +1,18 @@ +# @package _global_ +defaults: + - override /models: text-vae/hier_vae.py + - override /datasets: qm9_selfies + - override /losses: vae + + +experiment_name: "GR_061022_skOLI42s_vae_qm9_forward" +project: chem-vae +task: text-vae-forward +random_seed: 42 +accelerator: cpu +devices: 1 +batch_size: 128 +num_workers: 0 + +datasets: + work_mode: forward \ No newline at end of file diff --git a/config/experiments/demo_nm/GR_070922_jLqZn8Wh_catboost_regressor_mcl1.yaml b/config/experiments/demo_nm/GR_070922_jLqZn8Wh_catboost_regressor_mcl1.yaml new file mode 100644 index 00000000..524097e7 --- /dev/null +++ b/config/experiments/demo_nm/GR_070922_jLqZn8Wh_catboost_regressor_mcl1.yaml @@ -0,0 +1,10 @@ +# @package _global_ +defaults: + - override /models: regression/catboost_regression + - override /datasets: mcl1 + - override /callbacks: regression + +project: "mcl1" +task: "qsar-regression" +random_seed: 42 +# original_work_dir: ${hydra:runtime.cwd} diff --git a/config/experiments/demo_nm/GR_160822_AsdDwgf2_catboost_active_learning_qm9.yaml b/config/experiments/demo_nm/GR_160822_AsdDwgf2_catboost_active_learning_qm9.yaml new file mode 100644 index 00000000..70c5f474 --- /dev/null +++ b/config/experiments/demo_nm/GR_160822_AsdDwgf2_catboost_active_learning_qm9.yaml @@ -0,0 +1,16 @@ +# @package _global_ +defaults: + - override /models: regression/catboost_regression + - override /datasets: qm9 + - override /callbacks: regression + + +project: "qm9" +task: "qsar-regression" +random_seed: 42 + +extra: + active_learning: + query_size: 100 + epochs_num: 5 + use_data_uncertainty: False diff --git a/config/experiments/demo_nm/GR_180822_dAdsW3e5_catboost_uncertainty_active_learning_qm9.yaml b/config/experiments/demo_nm/GR_180822_dAdsW3e5_catboost_uncertainty_active_learning_qm9.yaml new file mode 100644 index 00000000..f03a85cd --- /dev/null +++ b/config/experiments/demo_nm/GR_180822_dAdsW3e5_catboost_uncertainty_active_learning_qm9.yaml @@ -0,0 +1,16 @@ +# @package _global_ +defaults: + - override /models: regression/catboost_regressor_uncertainty + - override /datasets: qm9 + - override /callbacks: regression + + +project: "qm9" +task: "qsar-regression" +random_seed: 42 + +extra: + active_learning: + query_size: 100 + epochs_num: 5 + use_data_uncertainty: True diff --git a/config/experiments/demo_nm/GR_230822_ASdw31ga_catboost_industry_data.yaml b/config/experiments/demo_nm/GR_230822_ASdw31ga_catboost_industry_data.yaml new file mode 100644 index 00000000..6957ea18 --- /dev/null +++ b/config/experiments/demo_nm/GR_230822_ASdw31ga_catboost_industry_data.yaml @@ -0,0 +1,11 @@ +# @package _global_ +defaults: + - override /models: regression/catboost_regression + - override /datasets: regression/regression_industry_data + - override /callbacks: regression + + +project: "industry_data" +task: "table-regression" +random_seed: 42 +#ckpt_path: https://api.blackhole.ai.innopolis.university/pretrained/catboost_industry_data.cbm diff --git a/config/experiments/demo_nm/GR_240822_BCXas132_catboost_regressor_qm9_screening.yaml b/config/experiments/demo_nm/GR_240822_BCXas132_catboost_regressor_qm9_screening.yaml new file mode 100644 index 00000000..99ceb395 --- /dev/null +++ b/config/experiments/demo_nm/GR_240822_BCXas132_catboost_regressor_qm9_screening.yaml @@ -0,0 +1,14 @@ +# @package _global_ +defaults: + - override /models: regression/catboost_regression + - override /datasets: qm9 + - override /callbacks: regression + + +project: "qm9" +task: "qsar-regression" +random_seed: 42 + +datasets: + target_value: 0.35 + delta_precision: 0.05 diff --git a/config/experiments/demo_nm/GR_300822_LIj29Hw_biobert_ner_drugprot.yaml b/config/experiments/demo_nm/GR_300822_LIj29Hw_biobert_ner_drugprot.yaml new file mode 100644 index 00000000..b2f6f0b5 --- /dev/null +++ b/config/experiments/demo_nm/GR_300822_LIj29Hw_biobert_ner_drugprot.yaml @@ -0,0 +1,17 @@ +# @package _global_ +defaults: + - override /models: classification/Bert_for_token_classification + - override /datasets: token_classification_drugprot + - override /losses: token_classification + + +project: biobert-ner +task: text-ner +random_seed: 42 +batch_size: 1 +weights_freq: 1 + +epochs: 1 + +trainer: + limit_train_batches: 1 \ No newline at end of file diff --git a/config/experiments/demo_nm/GR_300822_skOLIn2s_vae_qm9.yaml b/config/experiments/demo_nm/GR_300822_skOLIn2s_vae_qm9.yaml new file mode 100644 index 00000000..d6cb4008 --- /dev/null +++ b/config/experiments/demo_nm/GR_300822_skOLIn2s_vae_qm9.yaml @@ -0,0 +1,22 @@ +# @package _global_ +defaults: + - override /models: text-vae/hier_vae.py + - override /datasets: qm9_selfies + - override /losses: vae + + +experiment_name: "GR_300822_skOLIn2s_vae_qm9" +project: chem-vae +task: text-vae +random_seed: 42 +accelerator: cpu +devices: 1 +batch_size: 128 +epochs: 1 +num_workers: 0 + +trainer: + limit_train_batches: 10 + +datasets: + work_mode: vae \ No newline at end of file diff --git a/config/experiments/demo_nm/LR_150922_dAdeBe9_catboost_uncertainty_active_learning_mcl1.yaml b/config/experiments/demo_nm/LR_150922_dAdeBe9_catboost_uncertainty_active_learning_mcl1.yaml new file mode 100644 index 00000000..22fd7ec4 --- /dev/null +++ b/config/experiments/demo_nm/LR_150922_dAdeBe9_catboost_uncertainty_active_learning_mcl1.yaml @@ -0,0 +1,17 @@ +# @package _global_ +defaults: + - override /models: regression/catboost_regressor_uncertainty + - override /datasets: mcl1 + - override /callbacks: regression + + +project: "mcl1" +task: "table-regression" +random_seed: 42 + + +extra: + active_learning: + query_size: 100 + epochs_num: 5 + use_data_uncertainty: True \ No newline at end of file diff --git a/config/experiments/demo_prom/DA_230922_sdgh32lk_deeplab.yaml b/config/experiments/demo_prom/DA_230922_sdgh32lk_deeplab.yaml new file mode 100644 index 00000000..33c5c509 --- /dev/null +++ b/config/experiments/demo_prom/DA_230922_sdgh32lk_deeplab.yaml @@ -0,0 +1,23 @@ +# @package _global_ +defaults: + - override /models: semantic-segmentation/deeplabv3plus.yaml + - override /datasets: semantic-segmentation/segmentation_landslides + - override /optimizers: adam.yaml + - override /losses: segmentation_losses + - override /augmentations_train: none + - override /augmentations_val: none + - override /augmentations_test: none + - override /initializations: he + - override /callbacks: segmentation + +project: "segmentation" +task: "image-segmentation" + +random_seed: 42 +weights_freq: 1 +batch_size: 16 +epochs: 50 +accelerator: gpu +gpus: 2 #[2] +models: + in_channels: 4 diff --git a/config/experiments/demo_prom/DN_051923stroke.yaml b/config/experiments/demo_prom/DN_051923stroke.yaml new file mode 100644 index 00000000..1fad8277 --- /dev/null +++ b/config/experiments/demo_prom/DN_051923stroke.yaml @@ -0,0 +1,38 @@ +# @package _global_ +task: "image-segmentation" +defaults: + - override /models: semantic-segmentation/unet_smp.yaml + - override /datasets: semantic-segmentation/stroke_dataset.yaml + - override /losses: semantic-segmentation/focal_tversky_loss.yaml + - override /augmentations_train: stroke_segmentation.yaml + - override /augmentations_val: stroke_segmentation.yaml + - override /metrics: segmentation.yaml + - override /optimizers: lion.yaml + - override /loggers: wandb + +project: "segmentation" + +accelerator: gpu +random_seed: 42 +epochs: 50 +batch_size: 64 +# original_work_dir: ${hydra:runtime.cwd} +weights_freq: 1 +override hydra/job_logging: stdout +metrics: + num_classes: 1 + task: "image-segmentation" +models: + in_channels: 1 + classes: 1 + encoder_name: resnet34 + +optimizers: + lr: 3e-4 + +wandb: + enable: True + project: unet + group: innofw + job_type: training + tags: ['Unet', 'MSL', 'Lion', 'resnet34', '+InvertImg'] \ No newline at end of file diff --git a/config/experiments/demo_prom/DS_190423_8dca23dc_ucmerced.yaml b/config/experiments/demo_prom/DS_190423_8dca23dc_ucmerced.yaml new file mode 100644 index 00000000..fc65bdda --- /dev/null +++ b/config/experiments/demo_prom/DS_190423_8dca23dc_ucmerced.yaml @@ -0,0 +1,37 @@ +# @package _global_ +defaults: + - override /models: classification/resnet.yaml + - override /datasets: classification/classification_ucmerced.yaml + - override /augmentations_train: linear-roads-bin-seg + - override /augmentations_val: linear-roads-bin-seg + - override /augmentations_test: none + - override /losses: log_loss + - override /optimizers: adam + - override /trainer: auto + #- override /loggers: wandb + - override /callbacks: none + +models: + _target_: innofw.core.models.torch.architectures.classification.Resnet34 + num_classes: 21 + name: ucmerced_classification + description: resnet classification model trained on ucmerced dataset + ckpt_path: https://api.blackhole.ai.innopolis.university/pretrained/testing/satellite-classification.ckpt + + +project: "ucmerced_classification" +task: "image-classification" +random_seed: 42 +# original_work_dir: ${hydra:runtime.cwd} +weights_freq: 1 +batch_size: 256 +epochs: 30 +accelerator: gpu + +# wandb: +# enable: False +# entity: sadjava +# project: ucmerced +# group: 200423_just_testing_wandb +# job_type: training + diff --git a/config/experiments/demo_prom/IM_040822_rbe23lls_anomaly_detection_timeseries_lstm.yaml b/config/experiments/demo_prom/IM_040822_rbe23lls_anomaly_detection_timeseries_lstm.yaml new file mode 100644 index 00000000..9542d55f --- /dev/null +++ b/config/experiments/demo_prom/IM_040822_rbe23lls_anomaly_detection_timeseries_lstm.yaml @@ -0,0 +1,14 @@ +# @package _global_ +defaults: + - override /models: anomaly-detection/lstm_autoencoder.yaml + - override /datasets: anomaly_detection_timeseries_ecg.yaml + - override /losses: l1.yaml + - override /schedulers: cosine_annealing + +project: "ecg" +task: "anomaly-detection-timeseries" +random_seed: 42 +batch_size: 4 +weights_freq: 1 +epochs: 1 + diff --git a/config/experiments/demo_prom/IM_130124_gdsfjbjh2_siz.yaml b/config/experiments/demo_prom/IM_130124_gdsfjbjh2_siz.yaml new file mode 100644 index 00000000..62748c0b --- /dev/null +++ b/config/experiments/demo_prom/IM_130124_gdsfjbjh2_siz.yaml @@ -0,0 +1,13 @@ +# @package _global_ +accelerator: gpu +batch_size: 32 +ckpt_path: /home/qazybek/pmi_tatneft/innofw/dui/logs/something/20240122-093931/weights/best.pt +defaults: +- override /datasets: detection/detection_siz.yaml +- override /models: detection/yolov5.yaml +devices: 1 +epochs: 3 +project: siz +random_seed: 43 +task: image-detection +weights_freq: 1 diff --git a/config/experiments/demo_prom/IM_190722_gdsfjbjh2_yolov5_fruits.yaml b/config/experiments/demo_prom/IM_190722_gdsfjbjh2_yolov5_fruits.yaml new file mode 100644 index 00000000..d3b848c3 --- /dev/null +++ b/config/experiments/demo_prom/IM_190722_gdsfjbjh2_yolov5_fruits.yaml @@ -0,0 +1,10 @@ +# @package _global_ +batch_size: 2 +defaults: +- override /models: detection/yolov5 +- override /datasets: detection/detection_rotten_fruits +epochs: 3 +project: rotten_fruits +random_seed: 43 +task: image-detection +weights_freq: 1 diff --git a/config/experiments/demo_prom/IM_190722_vwer3f23_oneshotlearning.yaml b/config/experiments/demo_prom/IM_190722_vwer3f23_oneshotlearning.yaml new file mode 100644 index 00000000..05f3efda --- /dev/null +++ b/config/experiments/demo_prom/IM_190722_vwer3f23_oneshotlearning.yaml @@ -0,0 +1,18 @@ +# @package _global_ +defaults: + - override /models: one-shot-learning/one_shot_learning.yaml + - override /datasets: osl_faces.yaml + - override /augmentations_train: osl #one_shot_learning.yaml + - override /augmentations_val: osl + - override /augmentations_test: none + - override /losses: one_shot_learning.yaml + + +project: "faces" +task: "one-shot-learning" +random_seed: 42 +weights_freq: 1 +epochs: 50 +accelerator: "gpu" +devices: 1 +batch_size: 64 diff --git a/config/experiments/demo_prom/KA_031122_invalid_experiment.yaml b/config/experiments/demo_prom/KA_031122_invalid_experiment.yaml new file mode 100644 index 00000000..9201fbfa --- /dev/null +++ b/config/experiments/demo_prom/KA_031122_invalid_experiment.yaml @@ -0,0 +1,10 @@ +# @package _global_ +defaults: + - override /models: regression/linear_regression + - override /datasets: osl_faces + + +project: "incorrect" +task: "image-detection" +random_seed: 42 +weights_freq: 1 diff --git a/config/experiments/demo_prom/KA_081122_df82ds8_defaults.yaml b/config/experiments/demo_prom/KA_081122_df82ds8_defaults.yaml new file mode 100644 index 00000000..67f2ad58 --- /dev/null +++ b/config/experiments/demo_prom/KA_081122_df82ds8_defaults.yaml @@ -0,0 +1,11 @@ +# @package _global_ +defaults: + - override /datasets: regression/regression_house_prices + - override /models: regression/linear_regression + +project: "check_defaults" +task: "table-regression" +random_seed: 42 +epochs: 1 +batch_size: 2 +weights_freq: 1 diff --git a/config/experiments/demo_prom/KA_110923_r7fgu34_yolov5_aircraft_video_infer.yaml b/config/experiments/demo_prom/KA_110923_r7fgu34_yolov5_aircraft_video_infer.yaml new file mode 100644 index 00000000..16aa4c3c --- /dev/null +++ b/config/experiments/demo_prom/KA_110923_r7fgu34_yolov5_aircraft_video_infer.yaml @@ -0,0 +1,12 @@ +# @package _global_ +defaults: + - override /models: detection/yolov5 + - override /datasets: detection/air_data + - override /optimizers: sgd + +project: "air_data" +task: "image-detection" +random_seed: 42 +epochs: 2 +batch_size: 2 +weights_freq: 1 diff --git a/config/experiments/demo_prom/KA_120722_8adfcdaa_yolov5.yaml b/config/experiments/demo_prom/KA_120722_8adfcdaa_yolov5.yaml new file mode 100644 index 00000000..c883b532 --- /dev/null +++ b/config/experiments/demo_prom/KA_120722_8adfcdaa_yolov5.yaml @@ -0,0 +1,12 @@ +# @package _global_ +batch_size: 32 +defaults: +- override /models: detection/yolov5 +- override /datasets: detection_lep +- override /optimizers: sgd + +epochs: 500 +project: lep +random_seed: 42 +task: image-detection +weights_freq: 1 diff --git a/config/experiments/demo_prom/KA_120722_8adfcdaa_yolov5_non_pretrained.yaml b/config/experiments/demo_prom/KA_120722_8adfcdaa_yolov5_non_pretrained.yaml new file mode 100644 index 00000000..cfa98114 --- /dev/null +++ b/config/experiments/demo_prom/KA_120722_8adfcdaa_yolov5_non_pretrained.yaml @@ -0,0 +1,13 @@ +# @package _global_ +batch_size: 2 +ckpt_path: /home/qazybek/pmi_tatneft/innofw/dui/logs/something/20240122-082640/weights/best.pt +defaults: + - override /models: detection/yolov5 + - override /datasets: detection_lep + - override /initializations: random + +project: "lep" +task: "image-detection" + +random_seed: 42 +weights_freq: 1 diff --git a/config/experiments/demo_prom/KA_120722_8adfcdaa_yolov5_pretrained.yaml b/config/experiments/demo_prom/KA_120722_8adfcdaa_yolov5_pretrained.yaml new file mode 100644 index 00000000..f5c0dca8 --- /dev/null +++ b/config/experiments/demo_prom/KA_120722_8adfcdaa_yolov5_pretrained.yaml @@ -0,0 +1,15 @@ +# @package _global_ +defaults: + - override /models: detection/yolov5 + - override /datasets: detection_lep + + +project: "lep" +task: "image-detection" +random_seed: 42 +epochs: 1 +batch_size: 2 +weights_freq: 1 + + +ckpt_path: https://api.blackhole.ai.innopolis.university/pretrained/lep_yolov5.pt diff --git a/config/experiments/demo_prom/KA_120923_fad49tg_yolov5_sort_infer.yaml b/config/experiments/demo_prom/KA_120923_fad49tg_yolov5_sort_infer.yaml new file mode 100644 index 00000000..e654d2ec --- /dev/null +++ b/config/experiments/demo_prom/KA_120923_fad49tg_yolov5_sort_infer.yaml @@ -0,0 +1,13 @@ +# @package _global_ +defaults: + - override /models: detection/yolov5 + - override /datasets: detection/wares_sort + - override /optimizers: sgd + +project: "auto" +task: "image-detection" +random_seed: 42 +epochs: 2 +batch_size: 2 +weights_freq: 1 +accelerator: gpu \ No newline at end of file diff --git a/config/experiments/demo_prom/KA_120923_sa47f1v6_yolov5_auto_infer.yaml b/config/experiments/demo_prom/KA_120923_sa47f1v6_yolov5_auto_infer.yaml new file mode 100644 index 00000000..e74997f3 --- /dev/null +++ b/config/experiments/demo_prom/KA_120923_sa47f1v6_yolov5_auto_infer.yaml @@ -0,0 +1,12 @@ +# @package _global_ +defaults: + - override /models: detection/yolov5 + - override /datasets: detection/auto + - override /optimizers: sgd + +project: "auto" +task: "image-detection" +random_seed: 42 +epochs: 2 +batch_size: 2 +weights_freq: 1 diff --git a/config/experiments/demo_prom/KA_130722_9f7134db_linear_regression.yaml b/config/experiments/demo_prom/KA_130722_9f7134db_linear_regression.yaml new file mode 100644 index 00000000..9cfbb9dd --- /dev/null +++ b/config/experiments/demo_prom/KA_130722_9f7134db_linear_regression.yaml @@ -0,0 +1,10 @@ +# @package _global_ +defaults: + - override /models: regression/linear_regression + - override /datasets: regression/regression_house_prices + - override /callbacks: regression + + +project: "house_prices" +task: "table-regression" +random_seed: 42 diff --git a/config/experiments/demo_prom/KA_190722_0a70ef39_xgbregressor_cc.yaml b/config/experiments/demo_prom/KA_190722_0a70ef39_xgbregressor_cc.yaml new file mode 100644 index 00000000..e0e71914 --- /dev/null +++ b/config/experiments/demo_prom/KA_190722_0a70ef39_xgbregressor_cc.yaml @@ -0,0 +1,10 @@ +# @package _global_ +defaults: + - override /models: regression/xgboost_regressor + - override /datasets: regression/credit_cards + - override /callbacks: regression + + +project: "credit_cards" +task: "table-regression" +random_seed: 42 diff --git a/config/experiments/demo_prom/KA_230922_sdgh32lk_deeplab.yaml b/config/experiments/demo_prom/KA_230922_sdgh32lk_deeplab.yaml new file mode 100644 index 00000000..3d8621f5 --- /dev/null +++ b/config/experiments/demo_prom/KA_230922_sdgh32lk_deeplab.yaml @@ -0,0 +1,21 @@ +# @package _global_ +defaults: + - override /models: semantic-segmentation/deeplabv3plus.yaml + - override /datasets: semantic-segmentation/segmentation_arable + - override /optimizers: adam.yaml + - override /losses: semantic-segmentation/dice.yaml + - override /augmentations_train: arable_sentinel2 + - override /augmentations_val: arable_sentinel2 + - override /augmentations_test: arable_sentinel2 + - override /initializations: he + - override /callbacks: segmentation + - override /schedulers: lambda + +project: "segmentation" +task: "image-segmentation" +random_seed: 42 +weights_freq: 1 +batch_size: 10 +epochs: 3 +models: + in_channels: 4 diff --git a/config/experiments/demo_prom/KA_230922_sdgh32lk_deeplab_full.yaml b/config/experiments/demo_prom/KA_230922_sdgh32lk_deeplab_full.yaml new file mode 100644 index 00000000..e1ed485e --- /dev/null +++ b/config/experiments/demo_prom/KA_230922_sdgh32lk_deeplab_full.yaml @@ -0,0 +1,22 @@ +# @package _global_ +defaults: + - override /models: semantic-segmentation/deeplabv3plus.yaml + - override /datasets: semantic-segmentation/segmentation_arable_full + - override /optimizers: adam.yaml + - override /losses: segmentation_losses + - override /initializations: he + - override /augmentations_train: none #segmentation_satellite_imagery # _inference + - override /augmentations_val: none + - override /augmentations_test: none + - override /callbacks: segmentation + +project: "segmentation" +task: "image-segmentation" +random_seed: 42 +weights_freq: 1 +batch_size: 10 +epochs: 1 +accelerator: gpu +models: + in_channels: 4 + encoder_name: resnet101 diff --git a/config/experiments/demo_prom/KA_250123_439erfks_basic_water_erosion_segmentation.yaml b/config/experiments/demo_prom/KA_250123_439erfks_basic_water_erosion_segmentation.yaml new file mode 100644 index 00000000..577f3671 --- /dev/null +++ b/config/experiments/demo_prom/KA_250123_439erfks_basic_water_erosion_segmentation.yaml @@ -0,0 +1,39 @@ +# @package _global_ +defaults: + - override /models: semantic-segmentation/deeplabv3plus.yaml + - override /datasets: semantic-segmentation/segmentation_arable_full.yaml + - override /optimizers: adam.yaml + - override /losses: segmentation_losses # triple_segmentation_loss + - override /augmentations_train: none #segmentation_satellite_imagery + # - override /initializations: he + - override /callbacks: segmentation_2 + - override /wandb: default + +optimizers: + implementations: + torch: + Adam: + function: + lr: 3e-4 + +trainers: + overfit_batches: 20 # fit only 20 val batches and 40 train batches + +models: + encoder_name: resnet101 + +accelerator: gpu +devices: 1 +gpus: 1 # use second gpu + + +wandb: + project: water_erosion + group: 250123_just_testing_code + job_type: training + +project: "segmentation" +task: "image-segmentation" +random_seed: 42 +weights_freq: 1 +batch_size: 10 \ No newline at end of file diff --git a/config/experiments/demo_prom/KA_280722_9379c32c_kmeans_credit_cards.yaml b/config/experiments/demo_prom/KA_280722_9379c32c_kmeans_credit_cards.yaml new file mode 100644 index 00000000..8a6fdb22 --- /dev/null +++ b/config/experiments/demo_prom/KA_280722_9379c32c_kmeans_credit_cards.yaml @@ -0,0 +1,10 @@ +# @package _global_ +defaults: + - override /models: clustering/kmeans + - override /datasets: clustering/clustering_credit_cards + - override /callbacks: clustering + + +project: "credit_cards" +task: "table-clustering" +random_seed: 42 diff --git a/config/experiments/demo_prom/KA_280722_ba099979_catboost_regressor.yaml b/config/experiments/demo_prom/KA_280722_ba099979_catboost_regressor.yaml new file mode 100644 index 00000000..64e6e1a0 --- /dev/null +++ b/config/experiments/demo_prom/KA_280722_ba099979_catboost_regressor.yaml @@ -0,0 +1,10 @@ +# @package _global_ +defaults: + - override /models: regression/catboost_regression + - override /datasets: regression/regression_house_prices +# - override /callbacks: regression + + +project: "house_prices" +task: "table-regression" +random_seed: 42 diff --git a/config/experiments/demo_prom/KG_041223_ingieq921_unet.yaml b/config/experiments/demo_prom/KG_041223_ingieq921_unet.yaml new file mode 100644 index 00000000..58085b6a --- /dev/null +++ b/config/experiments/demo_prom/KG_041223_ingieq921_unet.yaml @@ -0,0 +1,21 @@ +# @package _global_ +defaults: + - override /models: semantic-segmentation/unet_smp.yaml + - override /datasets: semantic-segmentation/segmentation_arable.yaml + - override /optimizers: adam.yaml + - override /losses: segmentation_losses + +trainers: + overfit_batches: 20 # fit only 20 val batches and 40 train batches + +accelerator: gpu + +models: + in_channels: 4 + + +project: "segmentation" +task: "image-segmentation" +random_seed: 42 +weights_freq: 1 +batch_size: 10 \ No newline at end of file diff --git a/config/experiments/demo_prom/KG_090323_wpng92_mnist.yaml b/config/experiments/demo_prom/KG_090323_wpng92_mnist.yaml new file mode 100644 index 00000000..c0beebad --- /dev/null +++ b/config/experiments/demo_prom/KG_090323_wpng92_mnist.yaml @@ -0,0 +1,20 @@ +# @package _global_ +defaults: + - override /models: classification/resnet.yaml + - override /datasets: classification/classification_mnist.yaml + - override /augmentations_train: none #classification.yaml + - override /augmentations_val: none + - override /augmentations_test: none + - override /losses: log_loss.yaml + - override /optimizers: adam + - override /schedulers: + +project: "mnist_classification" +task: "image-classification" +random_seed: 42 +batch_size: 128 +epochs: 5 +accelerator: "gpu" +devices: 2 +num_workers: 1 +weights_freq: 7 diff --git a/config/experiments/demo_prom/KG_100323_ianfq90_lion_test.yaml b/config/experiments/demo_prom/KG_100323_ianfq90_lion_test.yaml new file mode 100644 index 00000000..2ff8ee0b --- /dev/null +++ b/config/experiments/demo_prom/KG_100323_ianfq90_lion_test.yaml @@ -0,0 +1,19 @@ +# @package _global_ +defaults: + - override /models: semantic-segmentation/unet + - override /datasets: semantic-segmentation/segmentation_brain.yaml + - override /optimizers: lion.yaml + - override /augmentations_train: none + - override /augmentations_val: none + - override /augmentations_test: none + - override /losses: segmentation_losses.yaml + +models: + in_channels: 1 +project: "segmentation" +task: "image-segmentation" +random_seed: 42 +stop_param: 10 +epochs: 10 +weights_freq: 1 +batch_size: 4 \ No newline at end of file diff --git a/config/experiments/demo_prom/KG_100323_lion_arable.yaml b/config/experiments/demo_prom/KG_100323_lion_arable.yaml new file mode 100644 index 00000000..c9ef4c34 --- /dev/null +++ b/config/experiments/demo_prom/KG_100323_lion_arable.yaml @@ -0,0 +1,19 @@ +# @package _global_ +batch_size: 4 +defaults: +- override /models: semantic-segmentation/unet +- override /datasets: semantic-segmentation/segmentation_arable.yaml +- override /optimizers: lion.yaml +- override /augmentations_train: none +- override /augmentations_val: none +- override /augmentations_test: none +- override /losses: segmentation_losses.yaml +- override /callbacks: segmentation_2 +epochs: 5 +models: + in_channels: 4 +project: segmentation +random_seed: 42 +stop_param: 10 +task: image-segmentation +weights_freq: 1 diff --git a/config/experiments/demo_prom/KG_190124_wpiig1_ttpla.yaml b/config/experiments/demo_prom/KG_190124_wpiig1_ttpla.yaml new file mode 100644 index 00000000..4e212bca --- /dev/null +++ b/config/experiments/demo_prom/KG_190124_wpiig1_ttpla.yaml @@ -0,0 +1,21 @@ +# @package _global_ +defaults: + - override /models: detection/yolov5 + - override /datasets: detection/ttpla + - override /optimizers: sgd + +project: "ttpla" +task: "image-detection" +random_seed: 42 +epochs: 20 +batch_size: 2 +weights_freq: 1 +accelerator: gpu +devices: 1 + +wandb: + enable: True + project: ttpla + entity: "k-galliamov" + group: none + job_type: training \ No newline at end of file diff --git a/config/experiments/demo_prom/KG_210124_xpibms_insplad.yaml b/config/experiments/demo_prom/KG_210124_xpibms_insplad.yaml new file mode 100644 index 00000000..d7d8aebe --- /dev/null +++ b/config/experiments/demo_prom/KG_210124_xpibms_insplad.yaml @@ -0,0 +1,21 @@ +# @package _global_ +defaults: + - override /models: detection/yolov5 + - override /datasets: detection_lep_insplad + - override /optimizers: sgd + +project: "ttpla" +task: "image-detection" +random_seed: 42 +epochs: 2 +batch_size: 2 +weights_freq: 1 +accelerator: gpu +devices: 1 + +wandb: + enable: True + project: ttpla + entity: "k-galliamov" + group: none + job_type: training \ No newline at end of file diff --git a/config/experiments/demo_prom/KG_220524_coco.yaml b/config/experiments/demo_prom/KG_220524_coco.yaml new file mode 100644 index 00000000..0c6b5dc5 --- /dev/null +++ b/config/experiments/demo_prom/KG_220524_coco.yaml @@ -0,0 +1,21 @@ +# @package _global_ +defaults: +- override /datasets: detection/coco_pose_recognition.yaml +- override /models: detection/yolo8_pose.yaml + +# devices: 1 +epochs: 3 +accelerator: gpu +batch_size: 32 +project: coco_pose_recognition +random_seed: 0 +task: image-detection +weights_freq: 1 + + +wandb: + enable: True + project: coco_pose_recognition + entity: "k-galliamov" + group: none + job_type: training \ No newline at end of file diff --git a/config/experiments/demo_prom/KG_230323_sjdv8f20_segformer.yaml b/config/experiments/demo_prom/KG_230323_sjdv8f20_segformer.yaml new file mode 100644 index 00000000..f9cba032 --- /dev/null +++ b/config/experiments/demo_prom/KG_230323_sjdv8f20_segformer.yaml @@ -0,0 +1,26 @@ +# @package _global_ +defaults: + - override /models: semantic-segmentation/segformer_for_segmentation.yaml + - override /datasets: semantic-segmentation/segmentation_brain.yaml + - override /augmentations_train: none + - override /optimizers: adam.yaml + - override /augmentations_val: none + - override /augmentations_test: none + - override /losses: segmentation_losses.yaml + - override /callbacks: segmentation + + +models: + num_channels: 1 + num_labels: 1 + retain_dim: True +project: "segmentation" +task: "image-segmentation" +random_seed: 42 +stop_param: 3 +epochs: 3 +weights_freq: 1 +batch_size: 8 +accelerator: cpu # gpu +#override hydra/job_logging: stdout + diff --git a/config/experiments/demo_prom/MK_ArTaxOr_yolov5.yaml b/config/experiments/demo_prom/MK_ArTaxOr_yolov5.yaml new file mode 100644 index 00000000..5c399e8e --- /dev/null +++ b/config/experiments/demo_prom/MK_ArTaxOr_yolov5.yaml @@ -0,0 +1,18 @@ +# @package _global_ +defaults: + - override /models: detection/yolov5 + - override /datasets: detection/detection_ar_tax_or + - override /optimizers: sgd + +optimizers: + lr0: 0.01 + lrf: 0.01 + +project: "ar_tax_or" +task: "image-detection" +random_seed: 42 +epochs: 40 +batch_size: 16 +weights_freq: 1 + +ckpt_path: https://api.blackhole.ai.innopolis.university/pretrained/testing/artaxor.pt \ No newline at end of file diff --git a/config/experiments/demo_prom/NP_010423_concatenated_classification.yaml b/config/experiments/demo_prom/NP_010423_concatenated_classification.yaml new file mode 100644 index 00000000..46fb36cc --- /dev/null +++ b/config/experiments/demo_prom/NP_010423_concatenated_classification.yaml @@ -0,0 +1,20 @@ +# @package _global_ +defaults: + # - /datasets@_dataset_dict.classification_mnist1: classification/classification_mnist + # - /datasets@_dataset_dict.classification_mnist2: classification/classification_mnist + - override /datasets: classification/classification_mnist + - override /models: classification/resnet.yaml + - override /augmentations_train: none #classification.yaml + - override /augmentations_val: none + - override /augmentations_test: none + - override /losses: log_loss.yaml + - override /optimizers: adam + +project: "mnist_classification" +task: "image-classification" +random_seed: 42 +# original_work_dir: ${hydra:runtime.cwd} +weights_freq: 1 +batch_size: 8 +stop_param: 1 +epochs: 1 diff --git a/config/experiments/demo_prom/NP_210523_yolov8_insects.yaml b/config/experiments/demo_prom/NP_210523_yolov8_insects.yaml new file mode 100644 index 00000000..8b3abf62 --- /dev/null +++ b/config/experiments/demo_prom/NP_210523_yolov8_insects.yaml @@ -0,0 +1,22 @@ +# @package _global_ +defaults: + # - override /models: ultralytics + - override /models: detection/yolov8 + - override /datasets: detection/detection_ar_tax_or + - override /loggers: wandb + - override /optimizers: sgd + +project: "insects" +task: "image-detection" +random_seed: 43 +epochs: 100 +batch_size: 4 +weights_freq: 1 +n_classes: 7 +imgsz: 640 +optimizers: + name: SGD +schedulers: + lr0: 0.0001 + lrf: 0.01 +#ckpt_path: 'C:\\Users\\nikit\\Downloads\\innofw-main\\innofw-main\\logs\\something\\20230522-125003\\weights\\best.pt'