Extensive library of computer models: from classic ResNet to the latest Swin transformer;
Metrics for evaluating computer vision models in TensorBoard and MLflow: classification, segmentation, metrics for finding similar images, for face recognition;
The single interface for loading and unloading computer models;
Convenient "packaging": TorchOK can be run through the Conda environment and in the cloud — using Docker containerizer or in Safe Maker on Amazon Web Services.
The wide selection of ready-made datasets: it is only necessary to prepare your data in the required format (CSV file with annotations and paths to images);
Modern infrastructure. TorchOK runs on machines with CPU, GPU, and also on multiple computers with multiple GPUs. There is support for TPU;
VLAD VINOGRADOV
Technical Director of EORA Data Lab
Our task is to bring “raw” computer models to production. Therefore, TorchOK will always have only the best technical solutions. Not the most accurate in terms of public benchmarks, but the most effective in terms of quality/speed.
Advantages of TorchOK
FOR DEVELOPERS
It's simple
Minimal knowledge in Deep Learning is enough to work in pipeline: prepare the dataset for training, train the neural network model and then transfer it for integration.
It is universal
Usually teams write their own pipeline for each project. We use the TorchOK codebase as the basis, and then modify it within the team.
It is suitable for collaboration
TorchOK can be cloned to work in several projects at the same time, for separate development teams. The errors and improvements found are eliminated and added to a single repository and are visible to everyone.
It is easily customized
You choose the training parameters of the neural network model: which network to use, which data to load, which metrics to count, how many GPUs you need.
FOR BUSINESS PROJECTS
Saving money: you have the finished result of many years of EORA work in your hands
Constantly updated database of neural network models
Only the best: we save that really works
Saving time: you can immediately start training your neural network
The most modern solutions in a simple "package"
Quality assurance
Yearly improvements since 2019
The current version is the third
Respected by the research community
Selection of the best technical solutions and programs
Team
Vyacheslav Shultz
Vladislav Vinogradov
Development of critical library blocks, integration of neural network architectures
Library core development, packaging, management
Correction of critical errors
Rashid Bayazitov
Adding of the new task types support
Aelita Shaikhutdinova
Roman Bogachev
Konstantin Kubrak
Transition to permanent support of key neural networks while maintaining the same type of structure: backbone, neck, head, hat
The ability to upload models to different frameworks: TensorRT, ONNX, OpenVINO
Ensuring compatibility of different frameworks and layers of neural network models