diff --git a/hydra_flash/conf/experiment/hymenoptera.yaml b/hydra_flash/conf/experiment/hymenoptera.yaml index 167fa08668eb2f8141329f2c537297b17800b136..730f180b09377ddda343152c609bdc9376596659 100644 --- a/hydra_flash/conf/experiment/hymenoptera.yaml +++ b/hydra_flash/conf/experiment/hymenoptera.yaml @@ -1,11 +1,11 @@ # @package _global_ defaults: - _self_ - - /server: local - /preprocess: hymenoptera - /datamodule: hymenoptera - /model: resnet18 - /datamodule@predict_datamodule: hymenoptera_test + - /server: local trainer: _target_: flash.Trainer diff --git a/hydra_flash/conf/experiment/lyft.yaml b/hydra_flash/conf/experiment/lyft.yaml index b12d382a4570f47721bfb4aacbfc8643fd7d2cfc..54a551426daa4ee829bc7da7467ce9ef4cc710e6 100644 --- a/hydra_flash/conf/experiment/lyft.yaml +++ b/hydra_flash/conf/experiment/lyft.yaml @@ -1,11 +1,11 @@ # @package _global_ defaults: - _self_ - - /server: local - /preprocess: lyft - /datamodule: lyft - /model: seg_mobilenet - /datamodule@predict_datamodule: lyft_test + - /server: local trainer: _target_: flash.Trainer diff --git a/hydra_flash/conf/experiment/movie_poster.yaml b/hydra_flash/conf/experiment/movie_poster.yaml index ba553527228eae64a8f9f82553d7638de5e2019c..4172b87bd94afe2f20bb991ec81d7c3bad01599c 100644 --- a/hydra_flash/conf/experiment/movie_poster.yaml +++ b/hydra_flash/conf/experiment/movie_poster.yaml @@ -1,11 +1,11 @@ # @package _global_ defaults: - _self_ - - /server: local - /preprocess: movie_poster - /datamodule: movie_posters - /model: resnet18 - /datamodule@predict_datamodule: movie_posters_test + - /server: local trainer: _target_: flash.Trainer