hook_outputs: Hook_outputs

View source: R/callbacks_hook.R

hook_outputsR Documentation

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


fastai documentation built on June 22, 2024, 11:15 a.m.