2024年11月9日
OFA-Sys/chinese-clip-vit-base-patch16 占用显存测试
摘要： model.get_image_features(inputs) 64 batch_size 2096MB 取消with torch.no_grad():后 8GB占满 16 batch_size 3886MB AutoModel .from_pretrained(MODEL_NAME) 执行慢，原 阅读全文
posted @ 2024-11-09 11:20 hrdom 阅读(227) 评论(0) 推荐(0)