You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I was getting RuntimeError: OrderedDict mutated during iteration.
It seems like using the same LightningModule object with ModelSummary and Trainer causes this error.
To Reproduce
frompytorch_lightning.core.memoryimportModelSummarymodel=CifarNet() # any pl module would work hereModelSummary(model,mode='full')
trainer=Trainer(fast_dev_run=True,gpus=1)
trainer.fit(model)
🐛 Bug
I was getting
RuntimeError: OrderedDict mutated during iteration
.It seems like using the same
LightningModule
object withModelSummary
andTrainer
causes this error.To Reproduce
Steps to reproduce the behavior:
ModelSummary
classtrainer.fit
with same object.Stacktrace
Expected behavior
We should be able to use same object with both the classes
Environment
The text was updated successfully, but these errors were encountered: