iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Upgrade LFNA

Browse Source
This commit is contained in:
D-X-Y
2021-05-13 13:42:33 +00:00
parent cfabd05de8
commit c2fa181bc5
3 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
exps/LFNA/lfna.py
View File

@@ -1,7 +1,7 @@
#####################################################
# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2021.04 #
#####################################################
# python exps/LFNA/lfna.py --env_version v1
# python exps/LFNA/lfna.py --env_version v1 --device cuda
#####################################################
import sys, time, copy, torch, random, argparse
from tqdm import tqdm

1
exps/LFNA/lfna_meta_model.py
View File

@@ -55,6 +55,7 @@ class LFNA_Meta(super_core.SuperModule):
order=super_core.LayerOrder.PostNorm,
)
)
layers.append(super_core.SuperLinear(time_embedding, time_embedding))
self.meta_corrector = super_core.SuperSequential(*layers)
model_kwargs = dict(