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
import torch
import os
import copy # Add this for deep copying
import copy
class AIIA(nn.Module):
def __init__(self, config: AIIAConfig, **kwargs):