dbpprt / u-2-net-portrait

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."
Apache License 2.0
117 stars 14 forks source link

training #9

Open ucb-pb opened 3 years ago

ucb-pb commented 3 years ago

Thank you for sharing your work! I'm trying to figure out how to start training, but am not very familiar with hydra and am running into errors with it. I modified conf/dataset.yaml, but also changed it back to what it was originally and it still gives the same error. Perhaps you might have an idea what the problem is?

 File "train.py", line 279, in <module>
    main()
  File "/usr/local/lib/python3.8/dist-packages/hydra/main.py", line 49, in decorated_main
    _run_hydra(
  File "/usr/local/lib/python3.8/dist-packages/hydra/_internal/utils.py", line 367, in _run_hydra
    run_and_report(
  File "/usr/local/lib/python3.8/dist-packages/hydra/_internal/utils.py", line 214, in run_and_report
    raise ex
  File "/usr/local/lib/python3.8/dist-packages/hydra/_internal/utils.py", line 211, in run_and_report
    return func()
  File "/usr/local/lib/python3.8/dist-packages/hydra/_internal/utils.py", line 368, in <lambda>
    lambda: hydra.run(
  File "/usr/local/lib/python3.8/dist-packages/hydra/_internal/hydra.py", line 110, in run
    _ = ret.return_value
  File "/usr/local/lib/python3.8/dist-packages/hydra/core/utils.py", line 233, in return_value
    raise self._return_value
  File "/usr/local/lib/python3.8/dist-packages/hydra/core/utils.py", line 160, in run_job
    ret.return_value = task_function(task_cfg)
  File "train.py", line 73, in main
    train_collate_fn = dataset.get_collate_fn(mode='train', channels_last=cfg.trainer.channels_last)
  File ".../u-2-net-portrait/lib/data/dataset.py", line 74, in get_collate_fn
    transforms[configuration.identifier] = [instantiate(step) for step in steps]
  File ".../u-2-net-portrait/lib/data/dataset.py", line 74, in <listcomp>
    transforms[configuration.identifier] = [instantiate(step) for step in steps]
  File "/usr/local/lib/python3.8/dist-packages/hydra/_internal/instantiate/_instantiate2.py", line 182, in instantiate
    raise InstantiationException(
hydra.errors.InstantiationException: Top level config has to be OmegaConf DictConfig, plain dict, or a Structured Config class or instance

As a side note, a requirements.txt file would be useful to include in this repo

Testhjf commented 2 years ago

@ucb-pb have you solve the problem?i have theproblem

gyupro commented 2 years ago

I solved this problem. Remove hydra and hydra-core and reinstall hydra-core by typing pip install hydra-core==1.1.0.dev3

dbpprt commented 2 years ago

There is an environment.yaml file in the repository to be used with conda as alternative for a requirements.txt