Skip to content

Commit

Permalink
Update Inception_ResNet_DenseNet.py
Browse files Browse the repository at this point in the history
reorder the import statement to fit the explanation flow
  • Loading branch information
alephpi authored Jan 21, 2024
1 parent 37e0798 commit 7c2f00e
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,6 @@

# %matplotlib inline
from IPython.display import HTML, display
from lightning.pytorch.callbacks import LearningRateMonitor, ModelCheckpoint
from PIL import Image
from torchvision import transforms
from torchvision.datasets import CIFAR10
Expand Down Expand Up @@ -295,6 +294,7 @@ def test_step(self, batch, batch_idx):

# %%
# Callbacks
from lightning.pytorch.callbacks import LearningRateMonitor, ModelCheckpoint

# %% [markdown]
# To allow running multiple different models with the same Lightning module, we define a function below that maps a model name to the model class.
Expand Down

0 comments on commit 7c2f00e

Please sign in to comment.