Runtime error when running stage3

[asrays]

Hi, thanks for the great work on this project. I was able to successfully train both Stage 1 and Stage 2, but I’m encountering an error when trying to run Stage 3.

From what I can tell, it seems that the model definition might be missing or not properly imported in train_stage3.py script.

Could you please check if the model definition for Stage 3 is included or guide me on where it should be located?

`
File "train_stage3.py", line 19, in
from models import Landmark_generator_v2_ICME as Landmark_transformer
ImportError: cannot import name 'Landmark_generator_v2_ICME' from 'models' (/home/ashray/STSA/models/init.py)

File "train_stage3.py", line 21, in
from models import Renderer_v3_ICME as Renderer
ImportError: cannot import name 'Renderer_v3_ICME' from 'models' (/home/ashray/STSA/models/init.py)
`

[AndrewDing616]

Sorry for that! We will fix it as soon as possible.