Hi,
Can someone demonstrate how to use the new checkpoint 148 feature for part of the model. For example how to use checkpoint for the _DenseBlock 28 of the  densenet implementation, since this layer stores intermediate feature maps and becomes memory intensive?
Should the checkpoint be called in the __init__ of the module or in forward?
Pointers would be really helpful.
Regards
Nabarun