weixin_43467572的博客

前言记录训练ViT遇到的问题。可学习参数from torchsummary import summaryvit_model = ViT( image_size = 256, patch_size = 32, num_classes = 1000, dim = 1024, depth = 6, heads = 16, mlp_dim = 2048, dropout = 0.1, emb_dropout = 0.1)devi

前言记录自己常用的指令Linuxvncserver :99vncserver kill:99查看当前运行的matlab进程ps -ef | grep MATLABkill/查看GPU的利用nvidia-smiconda安装packageconda activate basepip install XXXX

前言在自己的工作中遇到了要使用Pytorch训练神经网络但仿真程序却是在Matlab上的情况，为了得到一个快速的进展，就有了在Matlab调用Pytorch神经网络的想法。总的来说，还是比较好实现，但是作为新手难免会遇到些问题，所以就把自己的经验写在这篇帖子里，以便日后复习，也希望能给其他人一些思路。由于最近比较忙，可能更新的速度会比较慢，但是会一直更新完。平台和软件Matlab: 2021aPython: 3.8.3我是直接使用Anaconda下载的Python.要查看不同Matlab版本支持

环境配置Anaconda版本和升级conda -Vpython -Vconda update condaconda install python=3.Xconda install anaconda-clientconda update anaconda