Skip to content

yandex-cloud-examples/yc-datasphere-batch-execution

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 

Repository files navigation

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.