correct copying

This commit is contained in:
Falko Victor Habel 2025-01-26 13:09:23 +01:00
parent 32daaadddd
commit e5a5618160
1 changed files with 3 additions and 2 deletions

View File

@ -1,8 +1,9 @@
from config import AIIAConfig from .config import AIIAConfig
from torch import nn from torch import nn
import torch import torch
import os import os
import copy # Add this for deep copying import copy
class AIIA(nn.Module): class AIIA(nn.Module):
def __init__(self, config: AIIAConfig, **kwargs): def __init__(self, config: AIIAConfig, **kwargs):