ae body training

2023-04-07 22:11:43 +02:00
parent 280198038c
commit 345475a74f
3 changed files with 276 additions and 1 deletions
@@ -1,6 +1,6 @@
 import torch
 from torch import nn
-from models.convnext import ConvNeXt, CNBlock, CNBlockConfig
+from models.convnext_orig import ConvNeXt, CNBlock, CNBlockConfig
 from typing import Any, Callable, List, Optional, Sequence