hub / github.com/branislav1991/PyTorchProjectFramework

github.com/branislav1991/PyTorchProjectFramework @main

Chat with this repo

repository ↗ · DeepWiki ↗ · + Follow

83 symbols 226 edges 12 files 53 documented · 64% ● updated 6y ago★ 751 open issues

Browse by type

⨍ Functions 71 ◇ Types & classes 12

What it actually does AI analysis from the code graph — generated when you open this

loading…

Core symbols most depended-on inside this repo

↓ 3 filesBaseModel

This class is an abstract base class (ABC) for models.

models/base_model.py

↓ 2 filesVisualizer

This class includes several functions that can display images and print logging information.

utils/visualizer.py

↓ 2 filesparse_configuration

Loads config file if a string was passed and returns the input if a dictionary was passed.

utils/__init__.py

↓ 2 filescreate_model

Create a model given the configuration. This is the main interface between this package and train.py/validate.py

models/__init__.py

↓ 2 filescreate_dataset

Create a dataset given the configuration (loaded from the json file). This function wraps the class CustomDatasetDataLoader. This …

datasets/__init__.py

browse all functions →

Shape

Method 60

Class 12

Function 11

Languages

Python100%

Modules by API surface

models/segmentation_model.py25 symbols

models/base_model.py22 symbols

utils/visualizer.py9 symbols

datasets/__init__.py8 symbols

datasets/base_dataset.py7 symbols

utils/__init__.py4 symbols

datasets/segmentation2d_dataset.py4 symbols

models/__init__.py2 symbols

validate.py1 symbols

train.py1 symbols

For agents

$ claude mcp add PyTorchProjectFramework \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact

github.com/branislav1991/PyTorchProjectFramework @main

Core symbols most depended-on inside this repo

Shape

Languages

Modules by API surface

For agents

Ask about this repo answers extend the page