Need help training SAM2

Community Help
1

Hi all, I am currently following this blog post by @ James Gallagher (@rfjames ?)

Step 1 and 2 are done successfully, but I am stuck on step 3. When I run this in my Jupyter Notebook:

!python training/train.py -c 'configs/train.yaml' --use-cluster 0 --num-gpus 1

I get this error:

Traceback (most recent call last):
  File "/home2/minorai4/capstone/test/lucas4/sam2/training/train.py", line 270, in <module>
    main(args)
  File "/home2/minorai4/capstone/test/lucas4/sam2/training/train.py", line 124, in main
    cfg = compose(config_name=args.config)
          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home2/minorai4/.local/lib/python3.12/site-packages/hydra/compose.py", line 38, in compose
    cfg = gh.hydra.compose_config(
          ^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home2/minorai4/.local/lib/python3.12/site-packages/hydra/_internal/hydra.py", line 594, in compose_config
    cfg = self.config_loader.load_configuration(
          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home2/minorai4/.local/lib/python3.12/site-packages/hydra/_internal/config_loader_impl.py", line 142, in load_configuration
    return self._load_configuration_impl(
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home2/minorai4/.local/lib/python3.12/site-packages/hydra/_internal/config_loader_impl.py", line 243, in _load_configuration_impl
    self.ensure_main_config_source_available()
  File "/home2/minorai4/.local/lib/python3.12/site-packages/hydra/_internal/config_loader_impl.py", line 129, in ensure_main_config_source_available
    self._missing_config_error(
  File "/home2/minorai4/.local/lib/python3.12/site-packages/hydra/_internal/config_loader_impl.py", line 102, in _missing_config_error
    raise MissingConfigException(
hydra.errors.MissingConfigException: Primary config module 'sam2' not found.
Check that it's correct and contains an __init__.py file

I have no idea how to solve this issue as I have successfully followed every prior step in detail. Can anyone help me? Much appreciated.
I’m working in a Linux environment if that matters.

2

Same problem: python - SAM 2.1 what is causing hydra.errors.MissingConfigException: Primary config module 'sam2' not found? - Stack Overflow

3

I also tried this:

cp /home2/minorai4/capstone/test/lucas4/sam2/sam2/configs/train.yaml /home2/minorai4/.local/lib/python3.12/site-packages/sam2_configs

ls
__init__.py  __pycache__  sam2_hiera_b+.yaml  sam2_hiera_l.yaml  sam2_hiera_s.yaml  sam2_hiera_t.yaml  train.yaml