yimingz1218

Attention Heatmap Visualization

1

Hello, thank you for your amazing work! How can I visualize the attention heatmap for different body parts as well as the whole body? I think it should be under...

Vram needed when training

2

Hello, how much is the minimum required vram for the stage2 training process? I train it on an A100 80G with batch size 16, image size 512*1024, and it nearly...