hook_outputs {fastai}

Hook_outputs

Description

Return 'Hooks' that store activations of all 'modules' in 'self.stored'

Usage

hook_outputs(modules, detach = TRUE, cpu = FALSE, grad = FALSE)

Arguments

modules

modules

detach

detach or not

cpu

cpu or not

grad

grad or not

Value

None