Megvii-BaseDetection / Megvii-BaseDetection/YOLOX

Nano model training

Open
#902 8 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
10.6k
Forks
2.5k
PR merge metrics
No merged PRs in 30d

Description

hi there,
I am trying to train a "nano" model on custom dataset by using google Colab but I can't generate the train command because of "Exp" error.
note: I have checked the nano.py file and it is already updated
here is the message:
Traceback (most recent call last):
File "/content/YOLOX/yolox/exp/build.py", line 13, in get_exp_by_file
current_exp = importlib.import_module(os.path.basename(exp_file).split(".")[0])
File "/usr/lib/python3.7/importlib/init.py", line 127, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File "", line 1006, in _gcd_import
File "", line 983, in _find_and_load
File "", line 965, in _find_and_load_unlocked
ModuleNotFoundError: No module named 'nano'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "train.py", line 115, in
exp = get_exp(args.exp_file, args.name)
File "/content/YOLOX/yolox/exp/build.py", line 51, in get_exp
return get_exp_by_file(exp_file)
File "/content/YOLOX/yolox/exp/build.py", line 16, in get_exp_by_file
raise ImportError("{} doesn't contains class named 'Exp'".format(exp_file))
ImportError: nano.py doesn't contains class named 'Exp'

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start with yolox/exp/build.py and train.py to trace how the experiment file is imported and how the Exp class is checked; compare that path with the referenced nano.py file. Reproduce the command in the reported Google Colab setup, then verify that the nano experiment loads successfully and the training command can be generated.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.