models.nn.pyg_unet
¶
Module Contents¶
Classes¶
-
class
models.nn.pyg_unet.
UNet
(num_features, num_classes, hidden_size, num_layers, dropout)[source]¶ Bases:
models.BaseModel
models.nn.pyg_unet
¶models.nn.pyg_unet.
UNet
(num_features, num_classes, hidden_size, num_layers, dropout)[source]¶Bases: models.BaseModel