Li-Yuan Tsao

icon indicating a website link liyuantsao.github.io

icon company National Tsing Hua University icon location Hsinchu, Taiwan icon location Master's student @ Department of Computer Science, National Tsing Hua University.

Results 3 comments of Li-Yuan Tsao

HINT: We don't support broadcasting, please use `expand` yourself before calling `memory_efficient_attention`

Have the same question.

About the GPU memory usage

> I can train the model on 1 3090 with bs=1. But when I use 2 or more GPUs, the memory is not enough. +1

The PSNR and SSIM results

+1