Update LFNA test
This commit is contained in:
parent
ee5d8a8e21
commit
540b44367b
@ -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(
|
||||
|
Loading…
Reference in New Issue
Block a user