search

hunto / DIST_KD

Official implementation of paper "Knowledge Distillation from A Stronger Teacher", NeurIPS 2022
Apache License 2.0
138 stars 20 forks source link

ModuleNotFoundError: No module named 'dataset' #3

Closed zifuwanggg closed 2 years ago

zifuwanggg commented 2 years ago

When running train_kd.py:

Traceback (most recent call last):
  File "train_kd.py", line 28, in <module>
    from dataset.datasets import CSTrainValSet
hunto commented 2 years ago

Hi @zifuwanggg ,

Sorry, we forgot to add the dataset directory in the code, it has now been added to the latest code (https://github.com/hunto/DIST_KD/commit/e500a39290b4df82b32e8aed5863cd9501e04dab).

You can pull the update and this error should be fixed now :)

zifuwanggg commented 2 years ago

Thanks for the quick reply! Now it runs smoothly.