ae body training

This commit is contained in:
2023-04-07 22:11:43 +02:00
parent 280198038c
commit 345475a74f
3 changed files with 276 additions and 1 deletions
+1 -1
View File
@@ -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