File tree Expand file tree Collapse file tree 4 files changed +4
-4
lines changed
Expand file tree Collapse file tree 4 files changed +4
-4
lines changed Original file line number Diff line number Diff line change 1414 if : github.event.pull_request.draft == false
1515 timeout-minutes : 61 # in case something is wrong with the internal timeout
1616 steps :
17- - uses : Lightning-AI/probot@v5.4
17+ - uses : Lightning-AI/probot@v5.5
1818 env :
1919 GITHUB_TOKEN : ${{ secrets.GITHUB_TOKEN }}
2020 with :
Original file line number Diff line number Diff line change @@ -32,7 +32,7 @@ Let's use the `AutoEncoder` as a feature extractor in a separate model.
3232 class CIFAR10Classifier(LightningModule):
3333 def __init__(self):
3434 # init the pretrained LightningModule
35- self.feature_extractor = AutoEncoder.load_from_checkpoint(PATH)
35+ self.feature_extractor = AutoEncoder.load_from_checkpoint(PATH).encoder
3636 self.feature_extractor.freeze()
3737
3838 # the autoencoder outputs a 100-dim representation and CIFAR-10 has 10 classes
Original file line number Diff line number Diff line change @@ -209,7 +209,7 @@ Glossary
209209
210210.. displayitem ::
211211 :header: LightningModule
212- :description: A base class organizug your neural network module
212+ :description: A base class organizing your neural network module
213213 :col_css: col-md-12
214214 :button_link: ../common/lightning_module.html
215215 :height: 100
Original file line number Diff line number Diff line change 1- 2.6.0dev
1+ 2.6.0dev0
You can’t perform that action at this time.
0 commit comments