Resnet

Module: generative_models.core.layers.resnet

Source: generative_models/core/layers/resnet.py

Overview

ResNet block implementations for JAX/Flax models.

This module provides ResNet and Bottleneck block implementations using the latest Flax NNX API.

Classes

BottleneckBlock

class BottleneckBlock

ResNetBlock

class ResNetBlock

Functions

call

def __call__()

call

def __call__()

init

def __init__()

init

def __init__()

create_resnet_block

def create_resnet_block()

create_resnet_stage

def create_resnet_stage()

Module Statistics

• Classes: 2
• Functions: 6
• Imports: 3