hook_output {fastai}R 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


[Package fastai version 2.2.2 Index]