iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update LFNA test

Browse Source
This commit is contained in:
D-X-Y
2021-05-13 05:24:40 +00:00
parent ee5d8a8e21
commit 540b44367b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
exps/LFNA/lfna-test-hpnet.py
View File

@@ -1,8 +1,8 @@
##################################################### #####################################################
# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2021.04 # # Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2021.04 #
##################################################### #####################################################
# python exps/LFNA/lfna-test-hpnet.py --env_version v1 --hidden_dim 16 --layer_dim 32 --epochs 500000 --init_lr 0.02 # python exps/LFNA/lfna-test-hpnet.py --env_version v1 --hidden_dim 16 --layer_dim 32 --epochs 500000 --init_lr 0.01
# python exps/LFNA/lfna-test-hpnet.py --env_version v1 --hidden_dim 16 --layer_dim 32 --epochs 500000 --init_lr 0.02 --device cuda # python exps/LFNA/lfna-test-hpnet.py --env_version v1 --hidden_dim 16 --layer_dim 32 --epochs 500000 --init_lr 0.01 --device cuda
##################################################### #####################################################
import sys, time, copy, torch, random, argparse import sys, time, copy, torch, random, argparse
from tqdm import tqdm from tqdm import tqdm
@@ -115,7 +115,7 @@ def main(args):
lr_scheduler.step() lr_scheduler.step()
loss_meter.update(final_loss.item()) loss_meter.update(final_loss.item())
if iepoch % 100 == 0: if iepoch % 20 == 0:
logger.log( logger.log(
head_str head_str
+ " meta-loss: {:.4f} ({:.4f}) :: lr={:.5f}, batch={:}".format( + " meta-loss: {:.4f} ({:.4f}) :: lr={:.5f}, batch={:}".format(