hook_output: Hook_output

View source: R/callbacks_hook.R

hook_outputR Documentation

Hook_output

Description

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

Usage

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 June 22, 2024, 11:15 a.m.