Hook: Hook

Description Usage Arguments Details Value

View source: R/callbacks_hook.R

Description

Create a hook on 'm' with 'hook_func'.

Usage

1
2
3
4
5
6
7
8
Hook(
  m,
  hook_func,
  is_forward = TRUE,
  detach = TRUE,
  cpu = FALSE,
  gather = FALSE
)

Arguments

m

m aprameter

hook_func

hook function

is_forward

is_forward or not

detach

detach or not

cpu

cpu or not

gather

gather or not

Details

Hooks are functions you can attach to a particular layer in your model and that will be executed in the forward pass (for forward hooks) or backward pass (for backward hooks).

Value

None


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