model3.Model

class model3.Model(*args, **kwargs)

Model template

__init__(*args, **kwargs)

The default initialization. Not recommended to touch.

call(x, *args, **kwargs)

Default function for class callables. Not recommended to touch.

forward(x, *args, **kwargs)

Here is where the model logits locates.

Parameters:x – Input tensor or numpy array. The object will be automatically converted to tensor if the input is np.array. Note that other arrays in args or kwargs will not be auto-converted.

initialize(*args, **kwargs)

Initialization function. Set layers, sub-modules and parameters here.