UnetConfig

class tfts.models.unet.UnetConfig(units: int = 64, kernel_size: int = 2, depth: int = 2, pool_sizes: Tuple[int, int] = (2, 4), upsampling_factors: Tuple[int, int, int] = (2, 2, 2), num_attention_heads: int = 4, attention_probs_dropout_prob: float = 0.1, hidden_dropout_prob: float = 0.1, use_residual: bool = False, use_attention: bool = False, use_se: bool = False, use_layer_norm: bool = False, **kwargs)

Bases: BaseConfig

Inherited-members:

Methods

from_dict(config_dict)
from_json(json_file)
from_pretrained(pretrained_model_name_or_path)
save_pretrained(save_directory)
to_dict()
to_json(json_file)
update(config_dict)

Attributes

attribute_map
model_type