yucornetto/MGMatting

Anyone ran this project in Mac M1 - without Cuda support?

shiwanlin opened this issue · 0 comments

shiwanlin commented

The codes are hardcoded with Cuda, e.g. in infer.py:

# build model
model = networks.get_generator(encoder=CONFIG.model.arch.encoder, decoder=CONFIG.model.arch.decoder)
model.cuda()