Skip to content

Latest commit

 

History

History
9 lines (5 loc) · 920 Bytes

README.md

File metadata and controls

9 lines (5 loc) · 920 Bytes

This repository contains an example of how to use Yandex DataSphere. You can find the full tutorial at https://cloud.yandex.ru/docs/datasphere/tutorials/batch-code-execution.

Based on a use case of a simple convolutional neural network (CNN), the tutorial shows how to set up deployment of a model trained in Yandex DataSphere using Yandex Cloud Functions. The example is based on this PyTorch tutorial: https://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html.

my_nn_model.py: Neural network architecture.

train_classifier.ipynb: Download a training sample of the CIFAR10 dataset and train the neural network described in the my_nn_model.py file. The trained model's weights will be saved to the cifar_net.pth file in the project storage.

test_classifier.ipynb: Upload the model architecture and weights created during model training. The model is used for predictions based on the test sample.