Get Started
Blog
Docs
PyTorch Lightning Lightning Fabric TorchMetrics Lightning Flash Lightning Bolts
GitHub
Lightning AI

Table of Contents

stable

Getting started

Quick Start
Installation & Troubleshooting
Flash Zero
Flash in Production

Tutorials

Electricity Price Forecasting with N-BEATS

Guides

Training from scratch
Finetuning
Predictions (inference)
TorchScript JIT Support
Data
Registry
Flash Serve
Backbones and Heads
Optimization (Optimizers and Schedulers)
Formatting Classification Targets
Customizing Transforms
Remote Data Loading

Image and Video

Image Classification
Multi-label Image Classification
Image Embedder
Object Detection
Keypoint Detection
Instance Segmentation
Semantic Segmentation
Style Transfer
Video Classification

Audio

Audio Classification
Speech Recognition

Tabular

Tabular Classification
Tabular Forecasting

Text

Text Classification
Multi-label Text Classification
Question Answering
Summarization
Translation
Text Embedder

Point Cloud

Point Cloud Segmentation
Point Cloud Object Detection

Graph

Graph Classification
Graph Embedder

Integrations

Providers
BaaL
FiftyOne
IceVision
Learn2Learn
PyTorch Forecasting
VISSL

API Reference

flash
flash.core
flash.core.data
flash.core.serve
flash.image
flash.audio
flash.pointcloud
flash.tabular
flash.text
flash.video
flash.graph

Contributing a Task

Introduction / Set-up
The Data
The Backbones
The Task
Optional Extras
The Example
The Tests
The Docs

Community

Flash Governance | Persons of interest
Flash API Stability
Contributing
Changelog

Docs >
flash.core >
Freeze
Edit on GitHub

Shortcuts

Freeze¶

class flash.core.finetuning.Freeze(train_bn=True)[source]¶

Next

Previous

© Copyright 2020-2021, PyTorch Lightning. Revision a9cedb5a.

Built with Sphinx using a theme provided by Read the Docs.

Freeze

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. Read PyTorch Lightning's Privacy Policy.

Get Started
Blog
Ecosystem

PyTorch Lightning
Lightning Fabric
TorchMetrics
Fabric

Github
Lightning.ai