detector_benchmark.watermark.sir.transform_model

Classes

ResidualBlock
TransformModel

Module Contents

class detector_benchmark.watermark.sir.transform_model.ResidualBlock(dim)

Bases: torch.nn.Module

fc

relu

forward(x)

class detector_benchmark.watermark.sir.transform_model.TransformModel(num_layers=4, input_dim=1024, hidden_dim=500, output_dim=300)

Bases: torch.nn.Module

layers

forward(x)