forked from mrq/DL-Art-School
Enable memory checkpointing
This commit is contained in:
parent
5189b11dac
commit
8c469b8286
|
@ -11,7 +11,7 @@ from torch.utils.checkpoint import checkpoint
|
|||
|
||||
|
||||
# Set to true to relieve memory pressure by using torch.utils.checkpoint in several memory-critical locations.
|
||||
memory_checkpointing_enabled = False
|
||||
memory_checkpointing_enabled = True
|
||||
|
||||
|
||||
# VGG-style layer with Conv(stride2)->BN->Activation->Conv->BN->Activation
|
||||
|
|
Loading…
Reference in New Issue
Block a user