Issue with RF-DETR Training: Missing wandb Module

Feedback
1

Hello Roboflow Team,

I am currently facing an issue while training a model using the RF-DETR architecture on your platform. During the training process, I encountered the following error:

No module named ‘wandb’
Traceback (most recent call last):

  • File “/train/bin/run_and_catch_error.py”, line 56, in *
  • runpy.run_path(path_name=entries[TASK_TYPE], run_name=“main”)*
  • File “/usr/lib/python3.10/runpy.py”, line 289, in run_path*
  • return _run_module_code(code, init_globals, run_name,*
  • File “/usr/lib/python3.10/runpy.py”, line 96, in _run_module_code*
  • _run_code(code, mod_globals, init_globals,*
  • File “/usr/lib/python3.10/runpy.py”, line 86, in _run_code*
  • exec(code, run_globals)*
  • File “config/rfdetr_object_detection_config.py”, line 2, in *
  • from src.rfdetr.base import RFDETRObjectDetectionTrainer*
  • File “/train/src/rfdetr/base.py”, line 12, in *
  • from rfdetr.deploy.export import make_infer_image*
  • File “/usr/local/lib/python3.10/dist-packages/rfdetr/deploy/export.py”, line 32, in *
  • import wandb*
    ModuleNotFoundError: No module named ‘wandb’

It seems that the wandb module is missing from the environment, which is causing the training to fail. As I am using the platform’s training environment, I cannot install missing dependencies directly.

I would appreciate your assistance in resolving this issue.

Thank you for your help!