ModuleNotFoundError: No module named 'main_train'

[prsbsvrn]

ModuleNotFoundError: No module named 'main_train'

[Sina-Najafi]

Hi there! I hope you are doing ok.
here is my solution:
Write py_modules=["main_train", "main_eval", "main_conversion", "main_loss_landscape"],
in setup.py before entry_points as an argument in the setup() function. and then run pip install --editable . to reinstall the package with these changes.
Good Luck

[prsbsvrn]

Hi there! I hope you are doing ok. here is my solution: Write py_modules=["main_train", "main_eval", "main_conversion", "main_loss_landscape"], in setup.py before entry_points as an argument in the setup() function. and then run pip install --editable . to reinstall the package with these changes. Good Luck

Thank you