首页 > 分享 > MINet代码逻辑

MINet代码逻辑

一 主逻辑

1 main.py

2 (exp_name, arg_config, path_config)
arg_config是在config.py文件中设置的
3 Solver的导入位置:


->下面进入solver.py

a 这里network_lib是由下得来

b self.arg_dict['model']由下得来:

因此查下self.arg_dict['model']的config文件中的键值对即可

这里的MINet_VGG16来自于network文件夹中的MINet.py文件中的MINet_VGG16这个类

二 优化器逻辑

1 创建self.net用于优化器设置

2 创建优化器

3 设置调度安排(貌似可有可无)

3 设置amp加速

a 设置优化器梯度为0
b 输入数据读入gpu
c 网络预测
d 监督loss
e amp的loss反向传播

三 监督逻辑

sovler.py 里面的def train(self):里面的内容

train_preds是预测结果
trin_masks是标签
self.loss_funs是损失
这里是loss的设置 = BCE+CEL

注意GAN的第一轮循环这里要加GAN的 loss
因此要对get_total_loss做改进

 

相关知识

基于C#制作一个桌面宠物代码
农学院课程代码
2024年逻辑门电子元件投资分析
方舟生存进化圣诞节宠物代码大全 方舟圣诞节宠物代码是什么
远程宠物自动喂食系统代码
javascript宠物网站代码 宠物网页设计
java版本电子宠物游戏代码
请教高手:魔兽世界单机版宠物技能代码!
有谁知道qq宠物氨基酸饮料代码?
QQ宠物氨基酸的代码哪里有?

网址: MINet代码逻辑 https://m.mcbbbk.com/newsview224962.html

所属分类:萌宠日常
上一篇: 如何在EXCEL计算时间相减公式
下一篇: [宏] 求个宠物切换目标攻击宏