Hooks {fastai}

Hooks

Description

Create several hooks on the modules in 'ms' with 'hook_func'.

Usage

Hooks(ms, hook_func, is_forward = TRUE, detach = TRUE, cpu = FALSE)

Arguments

ms

ms parameter

hook_func

hook function

is_forward

is_forward or not

detach

detach or not

cpu

cpu or not

Value

None