Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/SagiPolaczek/NeuralSVG
/ types & classes
Types & classes
36 in github.com/SagiPolaczek/NeuralSVG
⨍
Functions
216
◇
Types & classes
36
↓ 7 callers
Class
GaussianEmbedding
Gaussian positional embedding for strokes.
src/models/nerf_mlp_multi.py:457
↓ 5 callers
Class
LayerNorm
Subclass torch's LayerNorm to handle fp16.
CLIP_/clip/model.py:186
↓ 3 callers
Class
CustomMLP
MLP with optional residual connections.
src/models/nerf_mlp_multi.py:604
↓ 2 callers
Class
Bottleneck
CLIP_/clip/model.py:11
↓ 2 callers
Class
Coach
Training orchestrator for NeuralSVG models. Handles: - Training loop execution - Optimization and scheduling - Checkpoint management
src/training/coach.py:37
↓ 2 callers
Class
SchedulerType
Learning rate scheduler types.
src/configs/optim_config.py:8
↓ 2 callers
Class
SketchLRLambda
Learning rate scheduler for sketch training. This class provides various learning rate scheduling functions: - Sinusoidal ramp with exponenti
src/training/coach.py:1043
↓ 2 callers
Class
Transformer
CLIP_/clip/model.py:274
↓ 1 callers
Class
AttentionPool2d
CLIP_/clip/model.py:70
↓ 1 callers
Class
CLIP
CLIP_/clip/model.py:354
↓ 1 callers
Class
CoachLogger
Training logger for NeuralSVG.
src/training/logging/coach_logger.py:23
↓ 1 callers
Class
GaussianEmbeddingRGB
Gaussian embedding for RGB colors.
src/models/nerf_mlp_multi.py:497
↓ 1 callers
Class
ModifiedResNet
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 callers
Class
MultiheadAttention
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 callers
Class
NerfMLPMulti
Multi-output MLP for vector graphics generation.
src/models/nerf_mlp_multi.py:20
↓ 1 callers
Class
PainterNerf
Neural renderer for vector graphics using NeRF architecture.
src/models/painter_nerf.py:27
↓ 1 callers
Class
QuickGELU
CLIP_/clip/model.py:195
↓ 1 callers
Class
ResidualAttentionBlock
CLIP_/clip/model.py:200
↓ 1 callers
Class
SDSLoss
Score Distillation Sampling loss.
src/training/losses.py:36
↓ 1 callers
Class
SketchModel
Base model class for NeuralSVG implementations.
src/models/model.py:19
↓ 1 callers
Class
VisualTransformer
CLIP_/clip/model.py:292
↓ 1 callers
Class
XDoG
src/models/attention_utils.py:160
↓ 1 callers
Class
_LinearWithBias
CLIP_/clip/auxilary.py:21
Class
DataConfig
Configuration for training and evaluation data.
src/configs/data_config.py:9
Class
LogConfig
Configuration for logging and experiment tracking.
src/configs/log_config.py:9
Class
LogConfig
Logging configuration.
src/configs/train_config.py:105
Class
ModelConfig
Model architecture and training configuration.
src/configs/model_config.py:9
Class
ModelConfig
Model configuration.
src/configs/train_config.py:54
Class
OptimConfig
Optimization configuration.
src/configs/train_config.py:28
Class
OptimConfig
Optimization configuration for training.
src/configs/optim_config.py:22
Class
Painter
Neural painter model for generating vector graphics from text.
src/models/painter.py:20
Class
SchedulerType
Learning rate scheduler types.
src/configs/train_config.py:14
Class
SimpleTokenizer
CLIP_/clip/simple_tokenizer.py:68
Class
TrainConfig
Main training configuration.
src/configs/train_config.py:133
Class
TrainStepsConfig
Configuration for training steps and intervals.
src/configs/train_steps_config.py:7
Class
TrainStepsConfig
Training steps configuration.
src/configs/train_config.py:122