Can't train the model. ModuleNotFoundError: No module named 'sparse_image_warp'

[julyanghar]

Traceback (most recent call last):
File "tools/run_net.py", line 6, in
from slowfast.utils.misc import launch_job
File "/root/auditory-slow-fast/slowfast/utils/misc.py", line 20, in
from slowfast.datasets.utils import pack_pathway_output
File "/root/auditory-slow-fast/slowfast/datasets/init.py", line 5, in
from .epickitchens import Epickitchens
File "/root/auditory-slow-fast/slowfast/datasets/epickitchens.py", line 14, in
from .spec_augment import combined_transforms
File "/root/auditory-slow-fast/slowfast/datasets/spec_augment.py", line 3, in
from sparse_image_warp import sparse_image_warp
ModuleNotFoundError: No module named 'sparse_image_warp'
Actually spec_augment.py and sparse_image_warp.py are both in the datasets folder.

[YJXWHU]

You can check whether you have done: "export PYTHONPATH=/path/to/auditory-slow-fast/slowfast:$PYTHONPATH" to add the workspaceFolder to the PYTHONPATH.