Dheemanth R Joshi

Retrieving the Trained Model

6

How can we get back our trained model once we train using the pipe object and Gpipe Scheduler as a normal nn.Module class?

Request for training examples using PipeStage and PipeSchedule

3

Would it be possible to provide a few examples on how to train a network using pipestage and pipeschedule? all the examples i have gone through so far are dedicated...

Unable to find the compile_stage library

4

How can i import the compile_stage library to initialize my stage? it is not found after installation. ```[tasklist] ### Tasks ```

TorchVision Installation issue

1

How do install torch vision supported by dev version of pytorch required to run Hippy?

[BUG]: AttributeError: 'GeminiDDP' object has no attribute 'module'

1

### Is there an existing issue for this bug? - [X] I have searched the existing issues ### 🐛 Describe the bug I'm trying to run Colossal AI's example on...