hook_output: Hook_output

Description Usage Arguments Value

View source: R/callbacks_hook.R

Description

Return a 'Hook' that stores activations of 'module' in 'self$stored'

Usage

1
hook_output(module, detach = TRUE, cpu = FALSE, grad = FALSE)

Arguments

module

module

detach

detach or not

cpu

cpu or not

grad

grad or not

Value

None


fastai documentation built on Oct. 25, 2021, 5:08 p.m.