avid_cma @ git+https://github.com/YingtianDt/AVID-CMA.git
torch
torchvision