Lightning-AI / dl-fundamentals

Deep Learning Fundamentals -- Code material and exercises
https://lightning.ai/pages/courses/deep-learning-fundamentals/
320 stars 149 forks source link

Unit05-8 Adding Functionalities with Callback #68

Open chanhdpham opened 1 year ago

chanhdpham commented 1 year ago

The slides seem wrong. The slides are inconsistent with the API document about the Callback class. In the slides: from pytorch_lightning.callbacks import Callback Class MyCustomCallback(Callback): ... In the "Callback API" document: Class MyCustomCallback(lightning.Callback): ... I defined MyCustomCallback class for exercise 2 and it give me runtime error at trainer = L.Trainer(... When I used lightning.Callback, it works.