Skip to content

anabatsh/Tensor_Train

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Tensor_Train_Net

data - данные

lib - основные библиотеки:

1. **tt_format** - описание класса TensorTrain
2. **tt_methods** - математические операции и функции для методов класса ТТ
3. **tt_math** - операции с объектами типа TT: пока что содержит только TT_SVD для разложения d-мерных и 2D-тензоров (матриц) в ТТ
4. **layers** - классы для полносвязного и сверточного слоя с ТТ-разложением 
5. **dataset** - загрузка датасетов
6. **train** - функции для обучения сетей

Notebooks:

main - обучение сети на MNIST/CIFAR-10, сравнение качетсва оригинальной сети и сети с ТТ-разложением

optimizer - настройка оптимизатора, обучение сети на простом синтетическом датасете

check_tt_methods - проверка корректности методов класса TT

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published