Zihang Li

Can I train this model on RTX 3090 GPU

2

Hi Justin, May I ask if I can train this model on RTX 3090 GPU? It seems that tensorflow1.8 doesn't work properly on RTX3090. Sincerely, Zihang

您好，我想问一下face_parsing可以多卡训练嘛？

May I ask how to obtain the focus parameter of a group of pictures?

Bugs found when I run python main.py: ValueError: Can't write images with one color channel.

1

Process Process-13: Traceback (most recent call last): File "/home/lzh/anaconda3/envs/SGL/lib/python3.7/multiprocessing/process.py", line 297, in _bootstrap self.run() File "/home/lzh/anaconda3/envs/SGL/lib/python3.7/multiprocessing/process.py", line 99, in run self._target(*self._args, **self._kwargs) File "/data1/lzh/workspace/retinal-image-segmentation-and-analysis/SGL-Retinal-Vessel-Segmentation/sgl_training/utility.py", line 139, in bg _target imageio.imwrite(filename,...