Skip to content

Commit

Permalink
isort fix
Browse files Browse the repository at this point in the history
  • Loading branch information
stepp1 committed May 25, 2023
1 parent 18fbec5 commit 94b091b
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion eqxvision/models/classification/vgg.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from typing import Any, Dict, List, Optional, Union, cast
from typing import Any, cast, Dict, List, Optional, Union

import equinox as eqx
import equinox.nn as nn
Expand All @@ -10,6 +10,7 @@

from ...utils import load_torch_weights


_cfgs: Dict[str, List[Union[str, int]]] = {
"A": [64, "M", 128, "M", 256, 256, "M", 512, 512, "M", 512, 512, "M"],
"B": [64, 64, "M", 128, 128, "M", 256, 256, "M", 512, 512, "M", 512, 512, "M"],
Expand Down

0 comments on commit 94b091b

Please sign in to comment.