
深度学习
专注Android黑科技
专注安卓黑科技研究
展开
-
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
–原创 2021-04-24 17:11:33 · 1612 阅读 · 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 · 1625 阅读 · 2 评论