- 博客(2)
- 收藏
- 关注
原创 TX2 Jetpack4.3安装pytorch1.4.0+torchvision0.5.0
标题TX2 Jetpack4.3安装pytorch1.4.0+torchvision0.5.01. 查看安装的jetpack版本sudo pip3 install jetson-statssudo jtop然后选择5-INFO(键盘按5) 就可以清楚看到 Jetpack 版本,还有其他的硬件信息;2. 根据jetpack版本查看支持的pytorch版本由于TX2是arm架构,所以在pytorch官网无法安装,目前清华源好像有1.8.0和1.8.1版本的pytorch(感谢苹果M1带动arm行业
2021-04-22 14:07:29
956
5
原创 keras 中 model.predect() 与model.predict_classes()的区别
keras 中 model.predect() 与model.predict_classes()的区别前言今天在写程序的时候,发现keras中model.predict()与model.predict_classes()返回的结果不一样,在此记录。代码如下(示例):y = model.predict_classes(x)y_2 = model.predict(x)y_2_pre = np.argmax(y_2, axis=1) # axis = 1是取行的最大值的索引,0是列的最大值的索引
2021-02-27 12:22:49
2848
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人