学习目标:
ViTGAN代码、车牌识别接口
学习时间:
3.10-3.17
学习产出:
一、ViTGAN代码
上周改的ViTGAN版本Bug有点多,这周一直在改Bug,现在进度卡在这个Bug。更新扩散强度时需要得到不同gpu上真实图片的损失,但ada_stats的结果一直为nan导致diffusion的扩散强度无法计算。Diffusion-GAN模型在训练时就使用了分布式计算,但是ViTGAN的代码在生成器和鉴别器里面才使用分布式,现在不知道该怎么对不同的gpu进行追踪获得不同gpu上的损失。
第一个版本的ViTGAN的FID一直降不下去了,在20k之后是一条直线。
车牌检测接口
这周学习了Flask,准备使用Flask部署模型。目前已经完成了一些准备工作。