Pomprogram的博客

硬件环境操作系统：CentOS 8.3显卡：RTX 3090软件环境Pytorch版本：1.8.1PyTorch的CUDA版本：11.1代码/数据准备报错处理（1）报错1：RuntimeError: output with shape [1，256,192] doesn't match the broadcast shape [3，256，192]解决方法：修改dataset.py 第23行代码。修改前：修改后：（2）报错2：RuntimeError: view size is.

如果要将我们写好的Python代码生成API接口时，我们需要借助Flask框架1. 安装Flaskpip install Flask2. 编写Flask应用（1）新建python文件 app.py（注意创建的Python文件名不能为flask.py，这会与Flask 本身发生冲突）（2）在新建的文件里写下如下内容：#app.pyfrom flask import Flaskapp = Flask(__name__)@app.route('/test',methos=['post'])