MCPcopy Create free account
hub / github.com/DSL-Lab/StreamSplat / set_grad_checkpointing

Method set_grad_checkpointing

model/transformer_utils.py:393–394  ·  view source on GitHub ↗
(self, flag=True)

Source from the content-addressed store, hash-verified

391 self.grad_checkpointing = False
392
393 def set_grad_checkpointing(self, flag=True):
394 self.grad_checkpointing = flag
395
396 def forward(self, x, condition=None):
397 for res_i, module in enumerate(self.resblocks):

Callers 3

__init__Method · 0.45
__init__Method · 0.45
set_grad_checkpointingMethod · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected