diff --git a/.gitignore b/.gitignore
index 230d18ef49e63fcdae6700560adc1519fc4607b4..7bbc71c09205c78d790739d246bbe4f9f1881c17 100644
--- a/.gitignore
+++ b/.gitignore
@@ -99,10 +99,3 @@ ENV/
# mypy
.mypy_cache/
-
-# tensorflow model files:
-model-*
-events.out*
-
-.idea/
-
diff --git a/pretrained_models/bair_action_free/model.savp.transformation.flow.last_frames.2.generate_scratch_image.false.batch_size.16/checkpoint b/pretrained_models/bair_action_free/model.savp.transformation.flow.last_frames.2.generate_scratch_image.false.batch_size.16/checkpoint
deleted file mode 100644
index 03990405d77f4e1d371136abd431a4d0cad5c124..0000000000000000000000000000000000000000
--- a/pretrained_models/bair_action_free/model.savp.transformation.flow.last_frames.2.generate_scratch_image.false.batch_size.16/checkpoint
+++ /dev/null
@@ -1,4 +0,0 @@
-model_checkpoint_path: "model-300000"
-all_model_checkpoint_paths: "model-290000"
-all_model_checkpoint_paths: "model-295000"
-all_model_checkpoint_paths: "model-300000"
diff --git a/pretrained_models/bair_action_free/model.savp.transformation.flow.last_frames.2.generate_scratch_image.false.batch_size.16/dataset_hparams.json b/pretrained_models/bair_action_free/model.savp.transformation.flow.last_frames.2.generate_scratch_image.false.batch_size.16/dataset_hparams.json
deleted file mode 100644
index d8bb73bbde8e5f68690bba56fc337141594aaf94..0000000000000000000000000000000000000000
--- a/pretrained_models/bair_action_free/model.savp.transformation.flow.last_frames.2.generate_scratch_image.false.batch_size.16/dataset_hparams.json
+++ /dev/null
@@ -1,10 +0,0 @@
-{
- "context_frames": 2,
- "crop_size": 0,
- "force_time_shift": false,
- "frame_skip": 0,
- "scale_size": 0,
- "sequence_length": 15,
- "time_shift": 3,
- "use_state": true
-}
\ No newline at end of file
diff --git a/pretrained_models/bair_action_free/model.savp.transformation.flow.last_frames.2.generate_scratch_image.false.batch_size.16/model_hparams.json b/pretrained_models/bair_action_free/model.savp.transformation.flow.last_frames.2.generate_scratch_image.false.batch_size.16/model_hparams.json
deleted file mode 100644
index 5ec9c699817171931e2d701d1d38a3fe0cf35c61..0000000000000000000000000000000000000000
--- a/pretrained_models/bair_action_free/model.savp.transformation.flow.last_frames.2.generate_scratch_image.false.batch_size.16/model_hparams.json
+++ /dev/null
@@ -1,90 +0,0 @@
-{
- "ablation_conv_rnn_norm": false,
- "acvideo_gan_weight": 0.0,
- "acvideo_vae_gan_weight": 0.0,
- "ae_l2_weight": 0.0,
- "batch_size": 16,
- "beta1": 0.9,
- "beta2": 0.999,
- "clip_length": 10,
- "context_frames": 2,
- "context_images_background": false,
- "conv_rnn": "lstm",
- "d_conditional": true,
- "d_downsample_layer": "conv_pool2d",
- "d_net": "legacy",
- "d_use_gt_inputs": true,
- "decay_steps": [
- 200000,
- 300000
- ],
- "dependent_mask": true,
- "dilation_rate": [
- 1,
- 1
- ],
- "downsample_layer": "conv_pool2d",
- "e_net": "n_layer",
- "end_lr": 0.0,
- "feature_l2_weight": 0.0,
- "first_image_background": true,
- "gan_feature_cdist_weight": 0.0,
- "gan_feature_l2_weight": 0.0,
- "gan_loss_type": "LSGAN",
- "gan_weight": 0.0,
- "generate_scratch_image": false,
- "image_gan_weight": 0.0,
- "image_sn_gan_weight": 0.0,
- "image_sn_vae_gan_weight": 0.0,
- "image_vae_gan_weight": 0.0,
- "kernel_size": [
- 5,
- 5
- ],
- "kl_anneal": "linear",
- "kl_anneal_k": -1.0,
- "kl_anneal_steps": [
- 50000,
- 100000
- ],
- "kl_weight": 0.001,
- "l1_weight": 1.0,
- "l2_weight": 0.0,
- "last_frames": 2,
- "lr": 0.001,
- "max_steps": 300000,
- "n_layers": 3,
- "ndf": 32,
- "nef": 64,
- "ngf": 32,
- "norm_layer": "instance",
- "num_samples": 8,
- "num_transformed_images": 4,
- "nz": 8,
- "prev_image_background": true,
- "repeat": 3,
- "rnn": "lstm",
- "schedule_sampling": "inverse_sigmoid",
- "schedule_sampling_k": 900.0,
- "schedule_sampling_steps": [
- 0,
- 100000
- ],
- "sequence_length": 15,
- "state_weight": 0.0001,
- "transformation": "flow",
- "tuple_gan_weight": 0.0,
- "tuple_vae_gan_weight": 0.0,
- "tv_weight": 0.0,
- "upsample_layer": "upsample_conv2d",
- "use_e_rnn": false,
- "use_rnn_z": true,
- "vae_gan_weight": 0.0,
- "vgg_cdist_weight": 0.0,
- "video_gan_weight": 0.0,
- "video_sn_gan_weight": 0.0,
- "video_sn_vae_gan_weight": 0.0,
- "video_vae_gan_weight": 0.0,
- "where_add": "all",
- "z_l1_weight": 0.0
-}
\ No newline at end of file
diff --git a/pretrained_models/bair_action_free/model.savp.transformation.flow.last_frames.2.generate_scratch_image.false.batch_size.16/options.json b/pretrained_models/bair_action_free/model.savp.transformation.flow.last_frames.2.generate_scratch_image.false.batch_size.16/options.json
deleted file mode 100644
index 9c9073888871dd6132d34ab03ed4c182c7ac82c7..0000000000000000000000000000000000000000
--- a/pretrained_models/bair_action_free/model.savp.transformation.flow.last_frames.2.generate_scratch_image.false.batch_size.16/options.json
+++ /dev/null
@@ -1,23 +0,0 @@
-{
- "checkpoint": null,
- "dataset": "cartgripper",
- "dataset_hparams": null,
- "dataset_hparams_dict": null,
- "eval_summary_freq": 0,
- "gif_freq": 0,
- "gpu_mem_frac": 0,
- "image_summary_freq": 5000,
- "input_dir": "data/cartgripper_updown_sact",
- "logs_dir": "logs/cartgripper/vae",
- "metrics_freq": 0,
- "model": "savp",
- "model_hparams": "transformation=flow,last_frames=2,generate_scratch_image=false,batch_size=16",
- "model_hparams_dict": "hparams/bair/ours_vae_l1/model_hparams.json",
- "output_dir": "logs/cartgripper/vae/model.savp.transformation.flow.last_frames.2.generate_scratch_image.false.batch_size.16",
- "progress_freq": 100,
- "resume": false,
- "save_freq": 5000,
- "seed": null,
- "summary_freq": 1000,
- "val_input_dirs": null
-}
\ No newline at end of file
diff --git a/pretrained_models/bair_action_free/ours_savp/checkpoint b/pretrained_models/bair_action_free/ours_savp/checkpoint
deleted file mode 100644
index 1baef2343699d9d0727fab169e6df3797b097347..0000000000000000000000000000000000000000
--- a/pretrained_models/bair_action_free/ours_savp/checkpoint
+++ /dev/null
@@ -1 +0,0 @@
-model_checkpoint_path: "model-300000"
diff --git a/pretrained_models/bair_action_free/ours_savp/dataset_hparams.json b/pretrained_models/bair_action_free/ours_savp/dataset_hparams.json
deleted file mode 100644
index bc600df30375ad1f29705a681208fee956231273..0000000000000000000000000000000000000000
--- a/pretrained_models/bair_action_free/ours_savp/dataset_hparams.json
+++ /dev/null
@@ -1,10 +0,0 @@
-{
- "context_frames": 2,
- "crop_size": 0,
- "force_time_shift": false,
- "frame_skip": 0,
- "scale_size": 0,
- "sequence_length": 12,
- "time_shift": 2,
- "use_state": false
-}
\ No newline at end of file
diff --git a/pretrained_models/bair_action_free/ours_savp/model_hparams.json b/pretrained_models/bair_action_free/ours_savp/model_hparams.json
deleted file mode 100644
index 6b2be2f00aa97738926a58956aa022b84fbf816a..0000000000000000000000000000000000000000
--- a/pretrained_models/bair_action_free/ours_savp/model_hparams.json
+++ /dev/null
@@ -1,89 +0,0 @@
-{
- "ablation_conv_rnn_norm": false,
- "acvideo_gan_weight": 0.0,
- "acvideo_vae_gan_weight": 0.0,
- "ae_l2_weight": 0.0,
- "batch_size": 16,
- "beta1": 0.5,
- "beta2": 0.999,
- "clip_length": 10,
- "context_frames": 2,
- "context_images_background": false,
- "conv_rnn": "lstm",
- "d_conditional": true,
- "d_downsample_layer": "conv_pool2d",
- "d_net": "legacy",
- "d_use_gt_inputs": true,
- "decay_steps": [
- 200000,
- 300000
- ],
- "dependent_mask": true,
- "dilation_rate": [
- 1,
- 1
- ],
- "downsample_layer": "conv_pool2d",
- "e_net": "n_layer",
- "end_lr": 0.0,
- "feature_l2_weight": 0.0,
- "first_image_background": true,
- "gan_feature_cdist_weight": 0.0,
- "gan_feature_l2_weight": 0.0,
- "gan_loss_type": "GAN",
- "gan_weight": 0.0,
- "generate_scratch_image": true,
- "image_gan_weight": 0.0,
- "image_sn_gan_weight": 0.0,
- "image_sn_vae_gan_weight": 0.0,
- "image_vae_gan_weight": 0.0,
- "kernel_size": [
- 5,
- 5
- ],
- "kl_anneal": "linear",
- "kl_anneal_k": -1.0,
- "kl_anneal_steps": [
- 50000,
- 100000
- ],
- "kl_weight": 0.1,
- "l1_weight": 100.0,
- "l2_weight": 0.0,
- "last_frames": 1,
- "lr": 0.0002,
- "n_layers": 3,
- "ndf": 32,
- "nef": 64,
- "ngf": 32,
- "norm_layer": "instance",
- "num_samples": 8,
- "num_transformed_images": 4,
- "nz": 8,
- "prev_image_background": true,
- "repeat": 2,
- "rnn": "lstm",
- "schedule_sampling": "inverse_sigmoid",
- "schedule_sampling_k": 900.0,
- "schedule_sampling_steps": [
- 0,
- 100000
- ],
- "sequence_length": 12,
- "state_weight": 0.0,
- "transformation": "cdna",
- "tuple_gan_weight": 0.0,
- "tuple_vae_gan_weight": 0.0,
- "tv_weight": 0.0,
- "upsample_layer": "upsample_conv2d",
- "use_e_rnn": false,
- "use_rnn_z": true,
- "vae_gan_weight": 0.0,
- "vgg_cdist_weight": 0.0,
- "video_gan_weight": 0.0,
- "video_sn_gan_weight": 1.0,
- "video_sn_vae_gan_weight": 1.0,
- "video_vae_gan_weight": 0.0,
- "where_add": "all",
- "z_l1_weight": 0.0
-}
\ No newline at end of file
diff --git a/pretrained_models/bair_action_free/ours_savp/options.json b/pretrained_models/bair_action_free/ours_savp/options.json
deleted file mode 100644
index d2ca36aef0f26fc998662242b19be7358779eac4..0000000000000000000000000000000000000000
--- a/pretrained_models/bair_action_free/ours_savp/options.json
+++ /dev/null
@@ -1,4 +0,0 @@
-{
- "dataset": "softmotion",
- "model": "savp"
-}
\ No newline at end of file
diff --git a/setup.py b/setup.py
deleted file mode 100644
index a7343dc1b060af5e047b337844e42f9daa853082..0000000000000000000000000000000000000000
--- a/setup.py
+++ /dev/null
@@ -1,8 +0,0 @@
-# setup.py
-from setuptools import setup
-
-setup(
- name='alex_video_prediction',
- version='0.1.0',
- packages=['video_prediction'],
-)
\ No newline at end of file