Python >= 3.6.0 required with all requirements.txt dependencies installed:
$ git clone https://github.com/software-homework-team/ViTorch.git
$ cd ViTorch/code
$ pip install -r requirements.txt
run the code with
$ sh run.sh
运行成功后将弹出ViTorch欢迎界面窗口,窗口上边从左到右分别是菜单按钮,各类网络层,和数据集按钮.
点击菜单按钮,左侧弹出界面选择拦,有Desinger和Train两个选项.
选择左侧的designer, 在上面可以选择神经网络层(如:激活函数,卷积,全连接等)来编辑神经网络
编辑完成后可以点击编辑窗口下的generate按键,从而生成对应的pytorch神经网络模型对象
同时,还可以选择需要使用的数据集,在数据集选择的界面,可以预览到数据集的图像和一些基本信息,可以进行选择并下载
这编辑完神经网络和选择了数据集之后,就可以开始训练神经网络模型了,可以在左侧的界面选择栏选择train,并且设置各种训练参数,接着就能开始训练了.
在训练和过程中还可以实时显示训练曲线,动态观察神经网络的训练效果.