hook_outputs {fastai} | R Documentation |
Return 'Hooks' that store activations of all 'modules' in 'self.stored'
hook_outputs(modules, detach = TRUE, cpu = FALSE, grad = FALSE)
modules |
modules |
detach |
detach or not |
cpu |
cpu or not |
grad |
grad or not |
None