深度学习
专注Android黑科技
专注安卓黑科技研究
展开
-
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
–原创 2021-04-24 17:11:33 · 1585 阅读 · 0 评论 -
Auto.js pytorch插件使用案列
第一次用,记录下 判断快手截图女生是否穿丝袜。 # 使用这句代码生成一下移动端模型即可 model = torch.jit.trace(model, example_input, strice=False) # 官网文档说这句可选,我这一执行这句生成的模型就load出错 # model = mobile_optimizer.optimize_for_mobile(mobile) model.save("siwa.plt") function log(str) { console.log(str);原创 2021-04-23 19:35:58 · 1502 阅读 · 2 评论