RadoslawCzubak / First_ML_Journey

Simple project to try and understand ML frameworks. (TensorFlow, PyTorch, PyTorch + Lightning)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Projekt: Rozpoznawanie ręcznie pisanych liczb z użyciem TensorFlow, PyTorch i PyTorch Lightning

Opis projektu (Project Description):

Projekt ten ma na celu przygotowanie modeli do rozpoznawania ręcznie pisanych liczb ze zbioru danych MNIST. Głównym celem projektu jest nauka i zrozumienie różnych frameworków do machine learningu, takich jak TensorFlow, PyTorch i PyTorch Lightning. Może zawierać błędy i niedoskonałości, które nie powinny być wykorzystywane w środowiskach produkcyjnych.

This project aims to prepare models for recognizing handwritten digits from the MNIST dataset. The main goal of the project is to learn and understand various machine learning frameworks, such as TensorFlow, PyTorch, and PyTorch Lightning. It may contain errors and imperfections that should not be used in production environments.

Struktura projektu (Project Structure):

Projekt składa się z następujących głównych części:

  • MNIST_tensorflow.ipynb - Notebook zawierający implementację modelu do rozpoznawania liczb przy użyciu TensorFlow.
  • MNIST.ipynb - Notebook zawierający implementację modelu do rozpoznawania liczb przy użyciu PyTorch.
  • MNIST_lightning.ipynb - Notebook zawierający implementację modelu do rozpoznawania liczb przy użyciu PyTorch z Lightning.
  • My images - Katalog z testowymi obrazami spoza zbioru danych

About

Simple project to try and understand ML frameworks. (TensorFlow, PyTorch, PyTorch + Lightning)


Languages

Language:Jupyter Notebook 100.0%