This commit is contained in:
D-X-Y 2021-05-28 02:22:59 +08:00
parent 36311809e8
commit 89448e433f
3 changed files with 9 additions and 9 deletions

View File

@ -1,9 +1,9 @@
#####################################################
# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2021.04 #
#####################################################
# python exps/GeMOSA/baselines/maml-ft.py --env_version v1 --hidden_dim 16 --inner_step 5
# python exps/GeMOSA/baselines/maml-ft.py --env_version v2 --hidden_dim 16 --inner_step 5
# python exps/GeMOSA/baselines/maml-ft.py --env_version v3 --hidden_dim 32 --inner_step 5
# python exps/GeMOSA/baselines/maml-ft.py --env_version v1 --hidden_dim 16 --inner_step 5 --device cuda
# python exps/GeMOSA/baselines/maml-ft.py --env_version v2 --hidden_dim 16 --inner_step 5 --device cuda
# python exps/GeMOSA/baselines/maml-ft.py --env_version v3 --hidden_dim 32 --inner_step 5 --device cuda
# python exps/GeMOSA/baselines/maml-ft.py --env_version v4 --hidden_dim 32 --inner_step 5 --device cuda
#####################################################
import sys, time, copy, torch, random, argparse

View File

@ -1,10 +1,10 @@
#####################################################
# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2021.04 #
#####################################################
# python exps/GeMOSA/baselines/maml-nof.py --env_version v1 --hidden_dim 16 --inner_step 5
# python exps/GeMOSA/baselines/maml-nof.py --env_version v2 --hidden_dim 16
# python exps/GeMOSA/baselines/maml-nof.py --env_version v3 --hidden_dim 32
# python exps/GeMOSA/baselines/maml-nof.py --env_version v4 --hidden_dim 32
# python exps/GeMOSA/baselines/maml-nof.py --env_version v1 --hidden_dim 16 --inner_step 5 --device cuda
# python exps/GeMOSA/baselines/maml-nof.py --env_version v2 --hidden_dim 16 --inner_step 5 --device cuda
# python exps/GeMOSA/baselines/maml-nof.py --env_version v3 --hidden_dim 32 --inner_step 5 --device cuda
# python exps/GeMOSA/baselines/maml-nof.py --env_version v4 --hidden_dim 32 --inner_step 5 --device cuda
#####################################################
import sys, time, copy, torch, random, argparse
from tqdm import tqdm

View File

@ -16,5 +16,5 @@ class TestSynethicEnv(unittest.TestCase):
for version in versions:
env = get_synthetic_env(version=version)
print(env)
for timestamp, tau in env:
self.assertEqual(tau.shape, (1000, env.ndim))
for timestamp, (x, y) in env:
self.assertEqual(x.shape, (1000, env.ndim))