diff --git a/README.md b/README.md index 60904636..76b81329 100644 --- a/README.md +++ b/README.md @@ -65,7 +65,7 @@ trainer.fit(model) ```python import torch.nn as nn -class ExampleModel(RootModule): +class My_Model(RootModule): def __init__(self): # define model