第二十八周周报

学习目标:

ViTGAN代码、车牌识别接口

学习时间:

3.10-3.17

学习产出:

一、ViTGAN代码

第二十八周周报_第1张图片
第二十八周周报_第2张图片

上周改的ViTGAN版本Bug有点多,这周一直在改Bug,现在进度卡在这个Bug。更新扩散强度时需要得到不同gpu上真实图片的损失,但ada_stats的结果一直为nan导致diffusion的扩散强度无法计算。Diffusion-GAN模型在训练时就使用了分布式计算,但是ViTGAN的代码在生成器和鉴别器里面才使用分布式,现在不知道该怎么对不同的gpu进行追踪获得不同gpu上的损失。
第二十八周周报_第3张图片
第一个版本的ViTGAN的FID一直降不下去了,在20k之后是一条直线。

车牌检测接口

这周学习了Flask,准备使用Flask部署模型。目前已经完成了一些准备工作。

视频

第二十八周周报_第4张图片

图像

第二十八周周报_第5张图片

你可能感兴趣的:(周报,人工智能)