随笔分类 -  ML

pytorch numpy 转换成 tensor -＞ torch.from_numpy() . torch.Tensor to numpy
摘要：sub_ts = torch.from_numpy(sub_img) #sub_img为numpy类型 如何从torch.Tensor转换成numpy呢？ img2 = img.numpy() 阅读全文
posted @ 2021-10-19 15:11 bH1pJ 阅读(13) 评论(0) 推荐(0) 编辑
ML多GPU训练时报错“'Not JSON Serializable:'”
摘要：坑：是不是因为多GPU训练的时候不能写JSON呢？ 我如何解决这个问题了？ #hist=UL_model.fit(imgs_train, labs_train, batch_size=Bch_Sz, nb_epoch=100, initial_epoch=0, validation_split=0. 阅读全文
posted @ 2019-10-08 11:15 bH1pJ 阅读(18) 评论(0) 推荐(0) 编辑
如何使用unet网络进行训练和预测——V2使用ShiXiong's Code
摘要：where is the code? Where is the main? How to run？ activate unet ipython notebook 阅读全文
posted @ 2019-09-20 14:27 bH1pJ 阅读(9) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示