动手学深度学习
Kadima°
行至水穷处,坐看云起时
展开
-
【Pytorch实现】——FLOPS,FLOPs,MACCs学习记录
FLOPS,FLOPs,MACCs学习记录我的模型有多快?——深度学习网络模型的运算复杂度、空间占用和内存访问情况计算用tensorflow计算模型复杂度(浮点计算量,FLOPs)PyTorch Tips(FLOPs计算/参数量/计算图可视化/相关性分析)CNN 模型所需的计算力(flops)和参数(parameters)数量是怎么计算的?...原创 2021-08-23 15:42:50 · 761 阅读 · 1 评论 -
TensorBoard学习记录
TensorBoard学习记录TensorBoard入门1-4TensorBoard的使用(一)TensorBoard的使用(二)TensorBoard的使用(三)TensorBoard的使用(四)总结:TensorBoard的使用一共可分成四步:设置保存路径:tenboard_dir = ‘./tensorboard/test1/’指定一个文件用来保存TensorFlow的图:writer = tf.summary.FileWriter(tenboard_dir + hparam)把原创 2021-08-23 10:43:58 · 199 阅读 · 0 评论 -
【Pytorch实现】——BN代码实现
BN代码实现import torchfrom torch import nndef batch_norm(X,gamma,beta,moving_mean,moving_var,eps,momentum): # 推理 if not torch.is_grad_enabled(): X_hat = (X - moving_mean) / torch.sqrt(moving_var + eps) # 训练 else: assert len(X.shape) in (2,4原创 2021-08-20 11:08:59 · 1778 阅读 · 0 评论