From 540b44367b8805c083edae7593e1da683fcb9ba1 Mon Sep 17 00:00:00 2001 From: D-X-Y <280835372@qq.com> Date: Thu, 13 May 2021 05:24:40 +0000 Subject: [PATCH] Update LFNA test --- exps/LFNA/lfna-test-hpnet.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/exps/LFNA/lfna-test-hpnet.py b/exps/LFNA/lfna-test-hpnet.py index 6b19462..2b815ed 100644 --- a/exps/LFNA/lfna-test-hpnet.py +++ b/exps/LFNA/lfna-test-hpnet.py @@ -1,8 +1,8 @@ ##################################################### # 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.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 +# 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 from tqdm import tqdm @@ -115,7 +115,7 @@ def main(args): lr_scheduler.step() loss_meter.update(final_loss.item()) - if iepoch % 100 == 0: + if iepoch % 20 == 0: logger.log( head_str + " meta-loss: {:.4f} ({:.4f}) :: lr={:.5f}, batch={:}".format(