pytorch
岳额不到五
这个作者很懒,什么都没留下…
展开
-
model.zero_grad()和optimizer.zero_grad()
搜了一堆博客,没看见个能解释明白的。一共就两三篇原创文章,其他人抄来抄去。可以参考这两篇:添加链接描述和添加链接描述。感觉就是model.zero_grad()将全部参数梯度置0;而optimizer.zero_grad()只将自己的参数梯度置0?...原创 2022-05-28 14:40:07 · 197 阅读 · 0 评论 -
如何计算模型running time,以及FLOPs
FLOPsFLOPs的定义及如何计算:添加链接描述Running time看这篇:添加链接描述原创 2022-05-27 16:37:23 · 391 阅读 · 0 评论 -
如何将多模态特征载入模型
《Learning a Text-Video Embedding fromIncomplete and Heterogeneous Data》所有数据存成.npy,然后np.load(.npy): class LSMDC(Dataset): def __init__(self, clip_path, text_features, audio_features, flow_path, face_path, **): self.visual_features = .原创 2021-06-02 16:30:52 · 788 阅读 · 0 评论