search

hpcaitech / ColossalAI-Examples

Examples of training models with hybrid parallelism using ColossalAI
Apache License 2.0
333 stars 102 forks source link

detr-debug pipelinable.py #197

Open lsx66 opened 1 year ago

lsx66 commented 1 year ago

https://github.com/hpcaitech/ColossalAI-Examples/tree/main/image/detr-debug

File" python3.8/site-packages/colossalai/pipeline/pipelinable.py", line 128, in to_layer_list layer_spec = self._layer_spec_dict[id(child)] KeyError: 140687581920992

binmakeswell commented 1 year ago

Hi @lsx66 Thank you for your feedback. I'm guessing it's due to an update to the pipeline. We will try to reproduce your issue and fix it soon.

By the way, we are restructuring the documents and examples, and the new version examples will be provided at the following link https://github.com/hpcaitech/ColossalAI/tree/main/examples