Skip to content
This repository was archived by the owner on Feb 7, 2025. It is now read-only.
This repository was archived by the owner on Feb 7, 2025. It is now read-only.

Fix autocast in 2d_ldm tutorial #149

@Warvito

Description

@Warvito

In the 2d LDM tutorial, the scalers for autocast are specified in

scaler_g = torch.cuda.amp.GradScaler()

but they were not used.
Consider add the autocast functionality or remove the scalers.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions