XY
This commit is contained in:
		| @@ -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 | ||||
|   | ||||
| @@ -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 | ||||
|   | ||||
| @@ -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)) | ||||
|   | ||||
		Reference in New Issue
	
	Block a user