Skip to content

Latest commit

 

History

History
23 lines (13 loc) · 1.16 KB

README.md

File metadata and controls

23 lines (13 loc) · 1.16 KB

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