head_index property

  1. @Int()
int head_index
getter/setter pair

The index of the classifier head (i.e. output tensor) these categories refer to. This is useful for multi-head models.

Implementation

@ffi.Int()
external int head_index;