ModuleNotFoundError: No module named 'training'

161 views Asked by At

Trying to train a new dataset from StyleGAN2 using stylegan2-ADA-pytorch using SageMaker Studio.

On the attempt to execute train.py, several imports fail. So I moved them out of train.py to call direct.

import os

import click import torch import json import tempfile import training

Error=


ModuleNotFoundError Traceback (most recent call last) Cell In[6], line 10 8 import json 9 import tempfile ---> 10 import training 12 from training import training_loop 13 from metrics import metric_main

ModuleNotFoundError: No module named 'training'

Training module is there. Was not an issue on Colab.

click and torch were also issues, but moving them out of train.py worked. Thoughts?

0

There are 0 answers