#想请问一下怎么在VGG网络中加入ECA模块啊,已经看视频学习了,但是人家是直接在进行卷积前进行模块的调用,但是我是跟着https://blog.csdn.net/weixin_44791964/article/details/109160814?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522168413512816800197022132%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fblog.%2522%257D&request_id=168413512816800197022132&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~blog~first_rank_ecpm_v1~rank_v31_ecpm-4-109160814-null-null.blog_rank_default&utm_term=VGG16&spm=1018.2226.3001.4450这位博主的内容进行的学习,在他的代码构筑中,直接选择了迭代,我不知道该怎么加入ECA模块了。想问问大家有什么办法。
将你的模块定义好,然后在forward函数中加入你的模块就行啊,代码循环的时候迭代的就是调用forward进行前向计算的啊,只要你的前向图里面有这个模块,训练的时候就会一起训练的,除非你将梯度计算关闭了,不然权重也会在反向计算的时候更新的