- 博客(2)
- 收藏
- 关注
RSS订阅原创 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
1041
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
2916
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人