-
与自动编码器潜在空间交互的visTorch
资源介绍
visTorch
visTorch是用于自动可视化自动编码器潜在空间的简单工具。 它是使用,并已通过Pytorch进行了测试。
安装
pip install -r requirements.txt
python setup.py install
用法
from visTorch import visboard
...
# your code
...
vis_board = visboard ()
vis_board . add_ae ( model ,
dataset ,
latent_options = { 'n' : model . latent_size , 'min' : - 1 , 'max' : 1 , 'step' : 0.05 },
model_paths = { 'Best'