├── .gitignore ├── LICENSE ├── README.md ├── configs ├── _base.yaml ├── collect_dataset_from_synthetic_envs.yaml ├── dataset │ ├── _base.yaml │ ├── _synthetic_env_base.yaml │ ├── cw-fingerimg-notarget-N4C11S1S1-hard.yaml │ ├── cw-notarget-N4C11S1S1-hard.yaml │ ├── odd-one-out-N4C2S2S1-oc.yaml │ ├── odd-one-out-N4C2S2S1.yaml │ ├── push-N3C4S1S1.yaml │ ├── random-N5C4S4S2.yaml │ └── target-N4C4S3S1.yaml ├── env │ ├── _cw_env_base.yaml │ ├── _synthetic_env_base.yaml │ ├── cw-fingerimg-notarget-N4C11S1S1-hard.yaml │ ├── cw-fingerimg-target-N4C4S1S1.yaml │ ├── cw-notarget-N4C11S1S1-hard.yaml │ ├── cw-ooo-N4C3S1S1.yaml │ ├── cw-target-N4C4S1S1-hard.yaml │ ├── cw-target-N4C4S1S1.yaml │ ├── maze-N0C4S4S2.yaml │ ├── maze-N4C4S4S2.yaml │ ├── odd-one-out-N3C2S2S1-oc.yaml │ ├── odd-one-out-N3C2S2S1.yaml │ ├── odd-one-out-N3C3S1S1.yaml │ ├── odd-one-out-N4C1S3S1.yaml │ ├── odd-one-out-N4C2S2S1-oc-ood-agent-color1.yaml │ ├── odd-one-out-N4C2S2S1-oc-ood-agent-color2.yaml │ ├── odd-one-out-N4C2S2S1-oc-ood-agent-shape1.yaml │ ├── odd-one-out-N4C2S2S1-oc-ood-agent-shape2.yaml │ ├── odd-one-out-N4C2S2S1-oc-ood-ocr-color1.yaml │ ├── odd-one-out-N4C2S2S1-oc-ood-ocr-color2.yaml │ ├── odd-one-out-N4C2S2S1-oc-ood-ocr-shape1.yaml │ ├── odd-one-out-N4C2S2S1-oc-ood-ocr-shape2.yaml │ ├── odd-one-out-N4C2S2S1-oc.yaml │ ├── odd-one-out-N4C2S2S1-ood-agent-color1.yaml │ ├── odd-one-out-N4C2S2S1-ood-agent-color2.yaml │ ├── odd-one-out-N4C2S2S1-ood-agent-shape1.yaml │ ├── odd-one-out-N4C2S2S1-ood-agent-shape2.yaml │ ├── odd-one-out-N4C2S2S1-ood-ocr-color1.yaml │ ├── odd-one-out-N4C2S2S1-ood-ocr-color2-shape2.yaml │ ├── odd-one-out-N4C2S2S1-ood-ocr-color2.yaml │ ├── odd-one-out-N4C2S2S1-ood-ocr-shape1.yaml │ ├── odd-one-out-N4C2S2S1-ood-ocr-shape2.yaml │ ├── odd-one-out-N4C2S2S1.yaml │ ├── odd-one-out-N4C2S2S2-oc.yaml │ ├── odd-one-out-N4C2S2S2.yaml │ ├── odd-one-out-N4C2S3S1-oc.yaml │ ├── odd-one-out-N4C2S3S1.yaml │ ├── odd-one-out-N4C2S3S2-oc.yaml │ ├── odd-one-out-N4C2S3S2.yaml │ ├── odd-one-out-N4C2S4S2-oc.yaml │ ├── odd-one-out-N4C2S4S2.yaml │ ├── odd-one-out-N4C3S1S1-ood-ocr-color1.yaml │ ├── odd-one-out-N4C3S1S1-ood-ocr-color2.yaml │ ├── odd-one-out-N4C3S1S1-ood-ocr-color3-shape1.yaml │ ├── odd-one-out-N4C3S1S1-ood-ocr-color3.yaml │ ├── odd-one-out-N4C3S1S1-ood-unseen-combi-test1.yaml │ ├── odd-one-out-N4C3S1S1-ood-unseen-combi-test2.yaml │ ├── odd-one-out-N4C3S1S1-ood-unseen-combi-test2_2.yaml │ ├── odd-one-out-N4C3S1S1-ood-unseen-combi-test3.yaml │ ├── odd-one-out-N4C3S1S1-ood-unseen-combi-test3_2.yaml │ ├── odd-one-out-N4C3S1S1-ood-unseen-combi-train1.yaml │ ├── odd-one-out-N4C3S1S1-ood-unseen-combi-train2.yaml │ ├── odd-one-out-N4C3S1S1-ood-unseen-combi-train3.yaml │ ├── odd-one-out-N4C3S1S1.yaml │ ├── odd-one-out-N4C3S2S1-oc.yaml │ ├── odd-one-out-N4C3S2S1.yaml │ ├── odd-one-out-N4C3S2S2-oc.yaml │ ├── odd-one-out-N4C3S2S2.yaml │ ├── odd-one-out-N4C3S3S2-oc.yaml │ ├── odd-one-out-N4C3S3S2.yaml │ ├── odd-one-out-N4C4S2S2-oc.yaml │ ├── odd-one-out-N4C4S2S2.yaml │ ├── odd-one-out-N4C4S3S2-oc.yaml │ ├── odd-one-out-N4C4S3S2.yaml │ ├── odd-one-out-N4C4S4S2.yaml │ ├── odd-one-out-N5C2S2S1-oc.yaml │ ├── odd-one-out-N5C2S2S1.yaml │ ├── odd-one-out-N5C3S1S1.yaml │ ├── odd-one-out-N6C2S2S1-oc.yaml │ ├── odd-one-out-N6C2S2S1-ood-ocr-color2-shape2.yaml │ ├── odd-one-out-N6C2S2S1.yaml │ ├── odd-one-out-N6C3S1S1-ood-ocr-color3-shape1.yaml │ ├── odd-one-out-N6C3S1S1.yaml │ ├── odd-one-out-N7C3S1S1.yaml │ ├── push-N1C1S1S1.yaml │ ├── push-N1C4S1S1.yaml │ ├── push-N2C3S1S1-ood-ocr-color1.yaml │ ├── push-N2C3S1S1-ood-ocr-color2.yaml │ ├── push-N2C3S1S1.yaml │ ├── push-N2C4S1S1.yaml │ ├── push-N3C3S1S1.yaml │ ├── push-N3C4S1S1-ood-ocr-color1.yaml │ ├── push-N3C4S1S1-ood-ocr-color2.yaml │ ├── push-N3C4S1S1-ood-ocr-color3.yaml │ ├── push-N3C4S1S1.yaml │ ├── push-N4C3S1S1.yaml │ ├── push-N4C4S1S1.yaml │ ├── push-N5C3S1S1.yaml │ ├── push-N5C4S1S1.yaml │ ├── push-N6C3S1S1.yaml │ ├── random-N1-9C4S4S2.yaml │ ├── random-N5C4S4S1.yaml │ ├── random-N5C4S4S2.yaml │ ├── target-N1C4S3S1.yaml │ ├── target-N2C3S3S1.yaml │ ├── target-N2C4S3S1.yaml │ ├── target-N3C3S3S1.yaml │ ├── target-N3C4S3S1.yaml │ ├── target-N4C3S3S1.yaml │ ├── target-N4C4S3S1-ood-ocr-color1.yaml │ ├── target-N4C4S3S1-ood-ocr-color2.yaml │ ├── target-N4C4S3S1-ood-ocr-color3.yaml │ ├── target-N4C4S3S1.yaml │ ├── target-N5C4S3S1.yaml │ ├── target-N6C4S3S1.yaml │ └── target-N7C4S3S1.yaml ├── get_ari_mse.yaml ├── ocr │ ├── gt.yaml │ ├── iodine_large.yaml │ ├── mae.yaml │ ├── multiple_cnn.yaml │ ├── naturecnn.yaml │ ├── slate.yaml │ └── vae.yaml ├── pooling │ ├── _base.yaml │ ├── cnn_linear.yaml │ ├── cnn_transformer.yaml │ ├── identity.yaml │ ├── mlp.yaml │ ├── rn.yaml │ └── transformer.yaml ├── sb3 │ ├── a2c.yaml │ ├── ppo.yaml │ └── sac.yaml ├── sb3_acnet │ ├── _base.yaml │ ├── identity.yaml │ ├── identity_orthoinit.yaml │ ├── mlp.yaml │ └── mlp_orthoinit.yaml ├── test_sb3.yaml ├── train_classifier.yaml ├── train_ocr.yaml ├── train_property_predictor.yaml ├── train_sb3.yaml └── wandb │ ├── _base.yaml │ ├── collect_dataset_from_synthetic_envs.yaml │ ├── get_ari_mse.yaml │ ├── offline.yaml │ ├── test_sb3.yaml │ ├── train_classifier.yaml │ ├── train_ocr.yaml │ ├── train_property_predictor.yaml │ └── train_sb3.yaml ├── envs ├── __init__.py ├── collect_dataset_from_cw_envs.py ├── cw_envs │ ├── __init__.py │ ├── cw.py │ └── target.py ├── manual_control.py └── synthetic_envs │ ├── __init__.py │ ├── base.py │ ├── maze.py │ ├── oddoneout.py │ ├── push.py │ ├── randomobjs.py │ ├── target.py │ └── utils.py ├── for_running.json ├── ocrs ├── __init__.py ├── base.py ├── common │ ├── models.py │ ├── networks.py │ ├── slot_attn.py │ ├── transformer.py │ ├── unet.py │ └── utils.py ├── gt │ ├── __init__.py │ ├── gt.py │ └── gt_module.py ├── iodine │ ├── __init__.py │ ├── iodine.py │ └── iodine_module.py ├── mae │ ├── __init__.py │ ├── mae.py │ ├── mae_module.py │ ├── models_mae.py │ ├── models_vit.py │ ├── util │ │ ├── crop.py │ │ ├── datasets.py │ │ ├── lars.py │ │ ├── lr_decay.py │ │ ├── lr_sched.py │ │ ├── misc.py │ │ └── pos_embed.py │ └── vit.py ├── multiple_cnns │ ├── __init__.py │ ├── multiple_cnn.py │ └── multiple_cnn_module.py ├── naturecnn │ ├── __init__.py │ ├── naturecnn.py │ └── naturecnn_module.py ├── slate │ ├── __init__.py │ ├── slate.py │ └── slate_module.py └── vaes │ ├── __init__.py │ ├── vae.py │ └── vae_module.py ├── poolings ├── __init__.py ├── base.py ├── cnn_linear │ ├── __init__.py │ ├── cnn_linear.py │ └── cnn_linear_module.py ├── cnn_transformer │ ├── __init__.py │ ├── cnn_transformer.py │ └── cnn_transformer_module.py ├── common │ ├── naturecnn.py │ └── transformer.py ├── identity │ ├── __init__.py │ ├── identity.py │ └── identity_module.py ├── mlp │ ├── __init__.py │ ├── mlp.py │ └── mlp_module.py ├── rn │ ├── __init__.py │ ├── rn.py │ └── rn_module.py └── transformer │ ├── __init__.py │ ├── transformer.py │ └── transformer_module.py ├── pretrained_encoders └── slate.pth ├── requirements.txt ├── run_sb3s.py ├── sb3s ├── __init__.py ├── custom_acnets.py └── ocr_extractor.py ├── scores ├── model_wise_performances.json ├── regime_wise_globalstep_wallclocktime.json └── regime_wise_performances.json ├── train_ocr.py ├── train_sb3.py └── utils ├── __init__.py ├── datasets.py ├── property_predictor.py └── tools.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/README.md -------------------------------------------------------------------------------- /configs/_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/_base.yaml -------------------------------------------------------------------------------- /configs/collect_dataset_from_synthetic_envs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/collect_dataset_from_synthetic_envs.yaml -------------------------------------------------------------------------------- /configs/dataset/_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/dataset/_base.yaml -------------------------------------------------------------------------------- /configs/dataset/_synthetic_env_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/dataset/_synthetic_env_base.yaml -------------------------------------------------------------------------------- /configs/dataset/cw-fingerimg-notarget-N4C11S1S1-hard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/dataset/cw-fingerimg-notarget-N4C11S1S1-hard.yaml -------------------------------------------------------------------------------- /configs/dataset/cw-notarget-N4C11S1S1-hard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/dataset/cw-notarget-N4C11S1S1-hard.yaml -------------------------------------------------------------------------------- /configs/dataset/odd-one-out-N4C2S2S1-oc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/dataset/odd-one-out-N4C2S2S1-oc.yaml -------------------------------------------------------------------------------- /configs/dataset/odd-one-out-N4C2S2S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/dataset/odd-one-out-N4C2S2S1.yaml -------------------------------------------------------------------------------- /configs/dataset/push-N3C4S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/dataset/push-N3C4S1S1.yaml -------------------------------------------------------------------------------- /configs/dataset/random-N5C4S4S2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/dataset/random-N5C4S4S2.yaml -------------------------------------------------------------------------------- /configs/dataset/target-N4C4S3S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/dataset/target-N4C4S3S1.yaml -------------------------------------------------------------------------------- /configs/env/_cw_env_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/_cw_env_base.yaml -------------------------------------------------------------------------------- /configs/env/_synthetic_env_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/_synthetic_env_base.yaml -------------------------------------------------------------------------------- /configs/env/cw-fingerimg-notarget-N4C11S1S1-hard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/cw-fingerimg-notarget-N4C11S1S1-hard.yaml -------------------------------------------------------------------------------- /configs/env/cw-fingerimg-target-N4C4S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/cw-fingerimg-target-N4C4S1S1.yaml -------------------------------------------------------------------------------- /configs/env/cw-notarget-N4C11S1S1-hard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/cw-notarget-N4C11S1S1-hard.yaml -------------------------------------------------------------------------------- /configs/env/cw-ooo-N4C3S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/cw-ooo-N4C3S1S1.yaml -------------------------------------------------------------------------------- /configs/env/cw-target-N4C4S1S1-hard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/cw-target-N4C4S1S1-hard.yaml -------------------------------------------------------------------------------- /configs/env/cw-target-N4C4S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/cw-target-N4C4S1S1.yaml -------------------------------------------------------------------------------- /configs/env/maze-N0C4S4S2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/maze-N0C4S4S2.yaml -------------------------------------------------------------------------------- /configs/env/maze-N4C4S4S2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/maze-N4C4S4S2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N3C2S2S1-oc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N3C2S2S1-oc.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N3C2S2S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N3C2S2S1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N3C3S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N3C3S1S1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C1S3S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C1S3S1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-oc-ood-agent-color1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-oc-ood-agent-color1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-oc-ood-agent-color2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-oc-ood-agent-color2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-oc-ood-agent-shape1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-oc-ood-agent-shape1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-oc-ood-agent-shape2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-oc-ood-agent-shape2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-oc-ood-ocr-color1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-oc-ood-ocr-color1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-oc-ood-ocr-color2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-oc-ood-ocr-color2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-oc-ood-ocr-shape1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-oc-ood-ocr-shape1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-oc-ood-ocr-shape2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-oc-ood-ocr-shape2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-oc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-oc.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-ood-agent-color1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-ood-agent-color1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-ood-agent-color2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-ood-agent-color2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-ood-agent-shape1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-ood-agent-shape1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-ood-agent-shape2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-ood-agent-shape2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-ood-ocr-color1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-ood-ocr-color1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-ood-ocr-color2-shape2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-ood-ocr-color2-shape2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-ood-ocr-color2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-ood-ocr-color2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-ood-ocr-shape1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-ood-ocr-shape1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1-ood-ocr-shape2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1-ood-ocr-shape2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S2-oc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S2-oc.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S2S2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S2S2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S3S1-oc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S3S1-oc.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S3S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S3S1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S3S2-oc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S3S2-oc.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S3S2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S3S2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S4S2-oc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S4S2-oc.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C2S4S2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C2S4S2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S1S1-ood-ocr-color1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S1S1-ood-ocr-color1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S1S1-ood-ocr-color2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S1S1-ood-ocr-color2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S1S1-ood-ocr-color3-shape1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S1S1-ood-ocr-color3-shape1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S1S1-ood-ocr-color3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S1S1-ood-ocr-color3.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S1S1-ood-unseen-combi-test1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S1S1-ood-unseen-combi-test1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S1S1-ood-unseen-combi-test2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S1S1-ood-unseen-combi-test2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S1S1-ood-unseen-combi-test2_2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S1S1-ood-unseen-combi-test2_2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S1S1-ood-unseen-combi-test3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S1S1-ood-unseen-combi-test3.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S1S1-ood-unseen-combi-test3_2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S1S1-ood-unseen-combi-test3_2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S1S1-ood-unseen-combi-train1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S1S1-ood-unseen-combi-train1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S1S1-ood-unseen-combi-train2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S1S1-ood-unseen-combi-train2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S1S1-ood-unseen-combi-train3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S1S1-ood-unseen-combi-train3.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S1S1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S2S1-oc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S2S1-oc.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S2S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S2S1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S2S2-oc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S2S2-oc.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S2S2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S2S2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S3S2-oc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S3S2-oc.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C3S3S2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C3S3S2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C4S2S2-oc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C4S2S2-oc.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C4S2S2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C4S2S2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C4S3S2-oc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C4S3S2-oc.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C4S3S2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C4S3S2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N4C4S4S2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N4C4S4S2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N5C2S2S1-oc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N5C2S2S1-oc.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N5C2S2S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N5C2S2S1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N5C3S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N5C3S1S1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N6C2S2S1-oc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N6C2S2S1-oc.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N6C2S2S1-ood-ocr-color2-shape2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N6C2S2S1-ood-ocr-color2-shape2.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N6C2S2S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N6C2S2S1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N6C3S1S1-ood-ocr-color3-shape1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N6C3S1S1-ood-ocr-color3-shape1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N6C3S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N6C3S1S1.yaml -------------------------------------------------------------------------------- /configs/env/odd-one-out-N7C3S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/odd-one-out-N7C3S1S1.yaml -------------------------------------------------------------------------------- /configs/env/push-N1C1S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/push-N1C1S1S1.yaml -------------------------------------------------------------------------------- /configs/env/push-N1C4S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/push-N1C4S1S1.yaml -------------------------------------------------------------------------------- /configs/env/push-N2C3S1S1-ood-ocr-color1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/push-N2C3S1S1-ood-ocr-color1.yaml -------------------------------------------------------------------------------- /configs/env/push-N2C3S1S1-ood-ocr-color2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/push-N2C3S1S1-ood-ocr-color2.yaml -------------------------------------------------------------------------------- /configs/env/push-N2C3S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/push-N2C3S1S1.yaml -------------------------------------------------------------------------------- /configs/env/push-N2C4S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/push-N2C4S1S1.yaml -------------------------------------------------------------------------------- /configs/env/push-N3C3S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/push-N3C3S1S1.yaml -------------------------------------------------------------------------------- /configs/env/push-N3C4S1S1-ood-ocr-color1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/push-N3C4S1S1-ood-ocr-color1.yaml -------------------------------------------------------------------------------- /configs/env/push-N3C4S1S1-ood-ocr-color2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/push-N3C4S1S1-ood-ocr-color2.yaml -------------------------------------------------------------------------------- /configs/env/push-N3C4S1S1-ood-ocr-color3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/push-N3C4S1S1-ood-ocr-color3.yaml -------------------------------------------------------------------------------- /configs/env/push-N3C4S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/push-N3C4S1S1.yaml -------------------------------------------------------------------------------- /configs/env/push-N4C3S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/push-N4C3S1S1.yaml -------------------------------------------------------------------------------- /configs/env/push-N4C4S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/push-N4C4S1S1.yaml -------------------------------------------------------------------------------- /configs/env/push-N5C3S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/push-N5C3S1S1.yaml -------------------------------------------------------------------------------- /configs/env/push-N5C4S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/push-N5C4S1S1.yaml -------------------------------------------------------------------------------- /configs/env/push-N6C3S1S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/push-N6C3S1S1.yaml -------------------------------------------------------------------------------- /configs/env/random-N1-9C4S4S2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/random-N1-9C4S4S2.yaml -------------------------------------------------------------------------------- /configs/env/random-N5C4S4S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/random-N5C4S4S1.yaml -------------------------------------------------------------------------------- /configs/env/random-N5C4S4S2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/random-N5C4S4S2.yaml -------------------------------------------------------------------------------- /configs/env/target-N1C4S3S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/target-N1C4S3S1.yaml -------------------------------------------------------------------------------- /configs/env/target-N2C3S3S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/target-N2C3S3S1.yaml -------------------------------------------------------------------------------- /configs/env/target-N2C4S3S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/target-N2C4S3S1.yaml -------------------------------------------------------------------------------- /configs/env/target-N3C3S3S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/target-N3C3S3S1.yaml -------------------------------------------------------------------------------- /configs/env/target-N3C4S3S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/target-N3C4S3S1.yaml -------------------------------------------------------------------------------- /configs/env/target-N4C3S3S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/target-N4C3S3S1.yaml -------------------------------------------------------------------------------- /configs/env/target-N4C4S3S1-ood-ocr-color1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/target-N4C4S3S1-ood-ocr-color1.yaml -------------------------------------------------------------------------------- /configs/env/target-N4C4S3S1-ood-ocr-color2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/target-N4C4S3S1-ood-ocr-color2.yaml -------------------------------------------------------------------------------- /configs/env/target-N4C4S3S1-ood-ocr-color3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/target-N4C4S3S1-ood-ocr-color3.yaml -------------------------------------------------------------------------------- /configs/env/target-N4C4S3S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/target-N4C4S3S1.yaml -------------------------------------------------------------------------------- /configs/env/target-N5C4S3S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/target-N5C4S3S1.yaml -------------------------------------------------------------------------------- /configs/env/target-N6C4S3S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/target-N6C4S3S1.yaml -------------------------------------------------------------------------------- /configs/env/target-N7C4S3S1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/env/target-N7C4S3S1.yaml -------------------------------------------------------------------------------- /configs/get_ari_mse.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/get_ari_mse.yaml -------------------------------------------------------------------------------- /configs/ocr/gt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/ocr/gt.yaml -------------------------------------------------------------------------------- /configs/ocr/iodine_large.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/ocr/iodine_large.yaml -------------------------------------------------------------------------------- /configs/ocr/mae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/ocr/mae.yaml -------------------------------------------------------------------------------- /configs/ocr/multiple_cnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/ocr/multiple_cnn.yaml -------------------------------------------------------------------------------- /configs/ocr/naturecnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/ocr/naturecnn.yaml -------------------------------------------------------------------------------- /configs/ocr/slate.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/ocr/slate.yaml -------------------------------------------------------------------------------- /configs/ocr/vae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/ocr/vae.yaml -------------------------------------------------------------------------------- /configs/pooling/_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/pooling/_base.yaml -------------------------------------------------------------------------------- /configs/pooling/cnn_linear.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/pooling/cnn_linear.yaml -------------------------------------------------------------------------------- /configs/pooling/cnn_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/pooling/cnn_transformer.yaml -------------------------------------------------------------------------------- /configs/pooling/identity.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/pooling/identity.yaml -------------------------------------------------------------------------------- /configs/pooling/mlp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/pooling/mlp.yaml -------------------------------------------------------------------------------- /configs/pooling/rn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/pooling/rn.yaml -------------------------------------------------------------------------------- /configs/pooling/transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/pooling/transformer.yaml -------------------------------------------------------------------------------- /configs/sb3/a2c.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/sb3/a2c.yaml -------------------------------------------------------------------------------- /configs/sb3/ppo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/sb3/ppo.yaml -------------------------------------------------------------------------------- /configs/sb3/sac.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/sb3/sac.yaml -------------------------------------------------------------------------------- /configs/sb3_acnet/_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/sb3_acnet/_base.yaml -------------------------------------------------------------------------------- /configs/sb3_acnet/identity.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/sb3_acnet/identity.yaml -------------------------------------------------------------------------------- /configs/sb3_acnet/identity_orthoinit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/sb3_acnet/identity_orthoinit.yaml -------------------------------------------------------------------------------- /configs/sb3_acnet/mlp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/sb3_acnet/mlp.yaml -------------------------------------------------------------------------------- /configs/sb3_acnet/mlp_orthoinit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/sb3_acnet/mlp_orthoinit.yaml -------------------------------------------------------------------------------- /configs/test_sb3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/test_sb3.yaml -------------------------------------------------------------------------------- /configs/train_classifier.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/train_classifier.yaml -------------------------------------------------------------------------------- /configs/train_ocr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/train_ocr.yaml -------------------------------------------------------------------------------- /configs/train_property_predictor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/train_property_predictor.yaml -------------------------------------------------------------------------------- /configs/train_sb3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/train_sb3.yaml -------------------------------------------------------------------------------- /configs/wandb/_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/wandb/_base.yaml -------------------------------------------------------------------------------- /configs/wandb/collect_dataset_from_synthetic_envs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/wandb/collect_dataset_from_synthetic_envs.yaml -------------------------------------------------------------------------------- /configs/wandb/get_ari_mse.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/wandb/get_ari_mse.yaml -------------------------------------------------------------------------------- /configs/wandb/offline.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/wandb/offline.yaml -------------------------------------------------------------------------------- /configs/wandb/test_sb3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/wandb/test_sb3.yaml -------------------------------------------------------------------------------- /configs/wandb/train_classifier.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/wandb/train_classifier.yaml -------------------------------------------------------------------------------- /configs/wandb/train_ocr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/wandb/train_ocr.yaml -------------------------------------------------------------------------------- /configs/wandb/train_property_predictor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/wandb/train_property_predictor.yaml -------------------------------------------------------------------------------- /configs/wandb/train_sb3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/configs/wandb/train_sb3.yaml -------------------------------------------------------------------------------- /envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/envs/__init__.py -------------------------------------------------------------------------------- /envs/collect_dataset_from_cw_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/envs/collect_dataset_from_cw_envs.py -------------------------------------------------------------------------------- /envs/cw_envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/envs/cw_envs/__init__.py -------------------------------------------------------------------------------- /envs/cw_envs/cw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/envs/cw_envs/cw.py -------------------------------------------------------------------------------- /envs/cw_envs/target.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/envs/cw_envs/target.py -------------------------------------------------------------------------------- /envs/manual_control.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/envs/manual_control.py -------------------------------------------------------------------------------- /envs/synthetic_envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/envs/synthetic_envs/__init__.py -------------------------------------------------------------------------------- /envs/synthetic_envs/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/envs/synthetic_envs/base.py -------------------------------------------------------------------------------- /envs/synthetic_envs/maze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/envs/synthetic_envs/maze.py -------------------------------------------------------------------------------- /envs/synthetic_envs/oddoneout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/envs/synthetic_envs/oddoneout.py -------------------------------------------------------------------------------- /envs/synthetic_envs/push.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/envs/synthetic_envs/push.py -------------------------------------------------------------------------------- /envs/synthetic_envs/randomobjs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/envs/synthetic_envs/randomobjs.py -------------------------------------------------------------------------------- /envs/synthetic_envs/target.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/envs/synthetic_envs/target.py -------------------------------------------------------------------------------- /envs/synthetic_envs/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/envs/synthetic_envs/utils.py -------------------------------------------------------------------------------- /for_running.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/for_running.json -------------------------------------------------------------------------------- /ocrs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/__init__.py -------------------------------------------------------------------------------- /ocrs/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/base.py -------------------------------------------------------------------------------- /ocrs/common/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/common/models.py -------------------------------------------------------------------------------- /ocrs/common/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/common/networks.py -------------------------------------------------------------------------------- /ocrs/common/slot_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/common/slot_attn.py -------------------------------------------------------------------------------- /ocrs/common/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/common/transformer.py -------------------------------------------------------------------------------- /ocrs/common/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/common/unet.py -------------------------------------------------------------------------------- /ocrs/common/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/common/utils.py -------------------------------------------------------------------------------- /ocrs/gt/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/gt/__init__.py -------------------------------------------------------------------------------- /ocrs/gt/gt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/gt/gt.py -------------------------------------------------------------------------------- /ocrs/gt/gt_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/gt/gt_module.py -------------------------------------------------------------------------------- /ocrs/iodine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/iodine/__init__.py -------------------------------------------------------------------------------- /ocrs/iodine/iodine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/iodine/iodine.py -------------------------------------------------------------------------------- /ocrs/iodine/iodine_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/iodine/iodine_module.py -------------------------------------------------------------------------------- /ocrs/mae/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/mae/__init__.py -------------------------------------------------------------------------------- /ocrs/mae/mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/mae/mae.py -------------------------------------------------------------------------------- /ocrs/mae/mae_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/mae/mae_module.py -------------------------------------------------------------------------------- /ocrs/mae/models_mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/mae/models_mae.py -------------------------------------------------------------------------------- /ocrs/mae/models_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/mae/models_vit.py -------------------------------------------------------------------------------- /ocrs/mae/util/crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/mae/util/crop.py -------------------------------------------------------------------------------- /ocrs/mae/util/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/mae/util/datasets.py -------------------------------------------------------------------------------- /ocrs/mae/util/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/mae/util/lars.py -------------------------------------------------------------------------------- /ocrs/mae/util/lr_decay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/mae/util/lr_decay.py -------------------------------------------------------------------------------- /ocrs/mae/util/lr_sched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/mae/util/lr_sched.py -------------------------------------------------------------------------------- /ocrs/mae/util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/mae/util/misc.py -------------------------------------------------------------------------------- /ocrs/mae/util/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/mae/util/pos_embed.py -------------------------------------------------------------------------------- /ocrs/mae/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/mae/vit.py -------------------------------------------------------------------------------- /ocrs/multiple_cnns/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/multiple_cnns/__init__.py -------------------------------------------------------------------------------- /ocrs/multiple_cnns/multiple_cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/multiple_cnns/multiple_cnn.py -------------------------------------------------------------------------------- /ocrs/multiple_cnns/multiple_cnn_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/multiple_cnns/multiple_cnn_module.py -------------------------------------------------------------------------------- /ocrs/naturecnn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/naturecnn/__init__.py -------------------------------------------------------------------------------- /ocrs/naturecnn/naturecnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/naturecnn/naturecnn.py -------------------------------------------------------------------------------- /ocrs/naturecnn/naturecnn_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/naturecnn/naturecnn_module.py -------------------------------------------------------------------------------- /ocrs/slate/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/slate/__init__.py -------------------------------------------------------------------------------- /ocrs/slate/slate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/slate/slate.py -------------------------------------------------------------------------------- /ocrs/slate/slate_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/slate/slate_module.py -------------------------------------------------------------------------------- /ocrs/vaes/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/vaes/__init__.py -------------------------------------------------------------------------------- /ocrs/vaes/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/vaes/vae.py -------------------------------------------------------------------------------- /ocrs/vaes/vae_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/ocrs/vaes/vae_module.py -------------------------------------------------------------------------------- /poolings/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/__init__.py -------------------------------------------------------------------------------- /poolings/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/base.py -------------------------------------------------------------------------------- /poolings/cnn_linear/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/cnn_linear/__init__.py -------------------------------------------------------------------------------- /poolings/cnn_linear/cnn_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/cnn_linear/cnn_linear.py -------------------------------------------------------------------------------- /poolings/cnn_linear/cnn_linear_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/cnn_linear/cnn_linear_module.py -------------------------------------------------------------------------------- /poolings/cnn_transformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/cnn_transformer/__init__.py -------------------------------------------------------------------------------- /poolings/cnn_transformer/cnn_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/cnn_transformer/cnn_transformer.py -------------------------------------------------------------------------------- /poolings/cnn_transformer/cnn_transformer_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/cnn_transformer/cnn_transformer_module.py -------------------------------------------------------------------------------- /poolings/common/naturecnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/common/naturecnn.py -------------------------------------------------------------------------------- /poolings/common/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/common/transformer.py -------------------------------------------------------------------------------- /poolings/identity/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/identity/__init__.py -------------------------------------------------------------------------------- /poolings/identity/identity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/identity/identity.py -------------------------------------------------------------------------------- /poolings/identity/identity_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/identity/identity_module.py -------------------------------------------------------------------------------- /poolings/mlp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/mlp/__init__.py -------------------------------------------------------------------------------- /poolings/mlp/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/mlp/mlp.py -------------------------------------------------------------------------------- /poolings/mlp/mlp_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/mlp/mlp_module.py -------------------------------------------------------------------------------- /poolings/rn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/rn/__init__.py -------------------------------------------------------------------------------- /poolings/rn/rn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/rn/rn.py -------------------------------------------------------------------------------- /poolings/rn/rn_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/rn/rn_module.py -------------------------------------------------------------------------------- /poolings/transformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/transformer/__init__.py -------------------------------------------------------------------------------- /poolings/transformer/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/transformer/transformer.py -------------------------------------------------------------------------------- /poolings/transformer/transformer_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/poolings/transformer/transformer_module.py -------------------------------------------------------------------------------- /pretrained_encoders/slate.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/pretrained_encoders/slate.pth -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_sb3s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/run_sb3s.py -------------------------------------------------------------------------------- /sb3s/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/sb3s/__init__.py -------------------------------------------------------------------------------- /sb3s/custom_acnets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/sb3s/custom_acnets.py -------------------------------------------------------------------------------- /sb3s/ocr_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/sb3s/ocr_extractor.py -------------------------------------------------------------------------------- /scores/model_wise_performances.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/scores/model_wise_performances.json -------------------------------------------------------------------------------- /scores/regime_wise_globalstep_wallclocktime.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/scores/regime_wise_globalstep_wallclocktime.json -------------------------------------------------------------------------------- /scores/regime_wise_performances.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/scores/regime_wise_performances.json -------------------------------------------------------------------------------- /train_ocr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/train_ocr.py -------------------------------------------------------------------------------- /train_sb3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/train_sb3.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/property_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/utils/property_predictor.py -------------------------------------------------------------------------------- /utils/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsikyoon/OCRL/HEAD/utils/tools.py --------------------------------------------------------------------------------