NetInfo

class lib.model.losses.feature_loss.NetInfo(model_id: int = 0, model_name: str = '', net: Callable | None = None, outputs: list[str] | list[int] = <factory>, pad_amount: list[int] | int = 0)

Bases: object

Data class for holding information about Trunk and Linear Layer nets.

Parameters:

• model_id (int) – The model ID for the model stored in the deepfakes Model repo

• model_name (str) – The filename of the decompressed model/weights file

• net (Callable | None) – The net definition to load, if any. Default:None

• outputs (list[str] | list[int]) – For trunk networks the name of the output feature layers. For linear networks the number of input channels to each layer

• pad_amount (list[int] | int) – For trunk networks, the amount of zero padding applied to each feature output

Attributes Summary

model_id
model_name
net
pad_amount

Attributes Documentation

model_id: int = 0

model_name: str = ''

net: Callable | None = None

pad_amount: list[int] | int = 0