MCPcopy Create free account

hub / github.com/SagiPolaczek/NeuralSVG / types & classes

Types & classes36 in github.com/SagiPolaczek/NeuralSVG

↓ 7 callersClassGaussianEmbedding
Gaussian positional embedding for strokes.
src/models/nerf_mlp_multi.py:457
↓ 5 callersClassLayerNorm
Subclass torch's LayerNorm to handle fp16.
CLIP_/clip/model.py:186
↓ 3 callersClassCustomMLP
MLP with optional residual connections.
src/models/nerf_mlp_multi.py:604
↓ 2 callersClassBottleneck
CLIP_/clip/model.py:11
↓ 2 callersClassCoach
Training orchestrator for NeuralSVG models. Handles: - Training loop execution - Optimization and scheduling - Checkpoint management
src/training/coach.py:37
↓ 2 callersClassSchedulerType
Learning rate scheduler types.
src/configs/optim_config.py:8
↓ 2 callersClassSketchLRLambda
Learning rate scheduler for sketch training. This class provides various learning rate scheduling functions: - Sinusoidal ramp with exponenti
src/training/coach.py:1043
↓ 2 callersClassTransformer
CLIP_/clip/model.py:274
↓ 1 callersClassAttentionPool2d
CLIP_/clip/model.py:70
↓ 1 callersClassCLIP
CLIP_/clip/model.py:354
↓ 1 callersClassCoachLogger
Training logger for NeuralSVG.
src/training/logging/coach_logger.py:23
↓ 1 callersClassGaussianEmbeddingRGB
Gaussian embedding for RGB colors.
src/models/nerf_mlp_multi.py:497
↓ 1 callersClassModifiedResNet
A ResNet class that is similar to torchvision's but contains the following changes: - There are now 3 "stem" convolutions as opposed to 1, wi
CLIP_/clip/model.py:116
↓ 1 callersClassMultiheadAttention
r"""Allows the model to jointly attend to information from different representation subspaces. See reference: Attention Is All You Need .
CLIP_/clip/auxilary.py:350
↓ 1 callersClassNerfMLPMulti
Multi-output MLP for vector graphics generation.
src/models/nerf_mlp_multi.py:20
↓ 1 callersClassPainterNerf
Neural renderer for vector graphics using NeRF architecture.
src/models/painter_nerf.py:27
↓ 1 callersClassQuickGELU
CLIP_/clip/model.py:195
↓ 1 callersClassResidualAttentionBlock
CLIP_/clip/model.py:200
↓ 1 callersClassSDSLoss
Score Distillation Sampling loss.
src/training/losses.py:36
↓ 1 callersClassSketchModel
Base model class for NeuralSVG implementations.
src/models/model.py:19
↓ 1 callersClassVisualTransformer
CLIP_/clip/model.py:292
↓ 1 callersClassXDoG
src/models/attention_utils.py:160
↓ 1 callersClass_LinearWithBias
CLIP_/clip/auxilary.py:21
ClassDataConfig
Configuration for training and evaluation data.
src/configs/data_config.py:9
ClassLogConfig
Configuration for logging and experiment tracking.
src/configs/log_config.py:9
ClassLogConfig
Logging configuration.
src/configs/train_config.py:105
ClassModelConfig
Model architecture and training configuration.
src/configs/model_config.py:9
ClassModelConfig
Model configuration.
src/configs/train_config.py:54
ClassOptimConfig
Optimization configuration.
src/configs/train_config.py:28
ClassOptimConfig
Optimization configuration for training.
src/configs/optim_config.py:22
ClassPainter
Neural painter model for generating vector graphics from text.
src/models/painter.py:20
ClassSchedulerType
Learning rate scheduler types.
src/configs/train_config.py:14
ClassSimpleTokenizer
CLIP_/clip/simple_tokenizer.py:68
ClassTrainConfig
Main training configuration.
src/configs/train_config.py:133
ClassTrainStepsConfig
Configuration for training steps and intervals.
src/configs/train_steps_config.py:7
ClassTrainStepsConfig
Training steps configuration.
src/configs/train_config.py:122