diff --git a/infer-web.py b/infer-web.py index 066ab54..daca13d 100644 --- a/infer-web.py +++ b/infer-web.py @@ -76,7 +76,8 @@ if torch.cuda.is_available() or ngpu != 0: "A3", "A4", "P4", - "A50","5000", + "A50", + "500", "A60", "70", "80",