gnes.encoder.image.torchvision module

class gnes.encoder.image.torchvision.TorchvisionEncoder(model_name, layers, model_dir, *args, **kwargs)

Bases: gnes.encoder.base.BaseImageEncoder

batch_size = 64

encode(img, *args, **kwargs)

Return type:ndarray

post_init()

Declare class attributes/members that can not be serialized in standard way

train(*args, **kwargs)

Train the model, need to be overrided