Skip to content
Snippets Groups Projects
Commit 4fd56d5e authored by Julien Dejasmin's avatar Julien Dejasmin
Browse files

update last 2

parent 31edd1e6
No related branches found
No related tags found
No related merge requests found
Showing
with 16 additions and 9220 deletions
/data1/home/julien.dejasmin/anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/_reduction.py:43: UserWarning: size_average and reduce args will be deprecated, please use reduction='sum' instead.
warnings.warn(warning.format(ret))
## OAR [2020-06-25 10:08:28] Job 2068292 KILLED ##
This diff is collapsed.
/data1/home/julien.dejasmin/anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/_reduction.py:43: UserWarning: size_average and reduce args will be deprecated, please use reduction='sum' instead.
warnings.warn(warning.format(ret))
This diff is collapsed.
/data1/home/julien.dejasmin/anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/_reduction.py:43: UserWarning: size_average and reduce args will be deprecated, please use reduction='sum' instead.
warnings.warn(warning.format(ret))
This diff is collapsed.
/data1/home/julien.dejasmin/anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/_reduction.py:43: UserWarning: size_average and reduce args will be deprecated, please use reduction='sum' instead.
warnings.warn(warning.format(ret))
This diff is collapsed.
...@@ -3,6 +3,7 @@ import numpy as np ...@@ -3,6 +3,7 @@ import numpy as np
from skimage.io import imread from skimage.io import imread
from torch.utils.data import Dataset, DataLoader, random_split from torch.utils.data import Dataset, DataLoader, random_split
from torchvision import datasets, transforms from torchvision import datasets, transforms
import os
# Load data # Load data
...@@ -108,6 +109,10 @@ def get_chairs_dataloader(num_worker=4, batch_size=128, path_to_data='../data/re ...@@ -108,6 +109,10 @@ def get_chairs_dataloader(num_worker=4, batch_size=128, path_to_data='../data/re
""" """
Chairs dataloader. Chairs are center cropped and resized to (64, 64). Chairs dataloader. Chairs are center cropped and resized to (64, 64).
""" """
path_to_data = 'data/rendered_chairs'
if not os.path.exists(path_to_data):
path_to_data = '../data/rendered_chairs'
all_transforms = transforms.Compose([ all_transforms = transforms.Compose([
# transforms.Grayscale(), # transforms.Grayscale(),
transforms.Resize(64), transforms.Resize(64),
......
...@@ -17,7 +17,6 @@ import json ...@@ -17,7 +17,6 @@ import json
def main(args): def main(args):
# continue and discrete capacity # continue and discrete capacity
if args.cont_capacity is not None: if args.cont_capacity is not None:
cont_capacity = [float(item) for item in args.cont_capacity.split(',')] cont_capacity = [float(item) for item in args.cont_capacity.split(',')]
...@@ -98,7 +97,6 @@ def main(args): ...@@ -98,7 +97,6 @@ def main(args):
args.latent_name + args.experiment_name + '.gif', viz)) args.latent_name + args.experiment_name + '.gif', viz))
if __name__ == "__main__": if __name__ == "__main__":
parser = argparse.ArgumentParser(description='VAE') parser = argparse.ArgumentParser(description='VAE')
parser.add_argument('--batch-size', type=int, default=64, metavar='integer value', parser.add_argument('--batch-size', type=int, default=64, metavar='integer value',
......
--batch-size=256 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=10 --is-beta-VAE=True --beta=4 --lr=1e-4 --experiment-name=beta_VAE_bs_256 --gpu-devices 0 1 --load-model-checkpoint=True --batch-size=256 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=10 --is-beta-VAE=True --beta=4 --lr=1e-4 --experiment-name=beta_VAE_bs_256 --gpu-devices 0 1 --load-model-checkpoint=True
--batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=10 --is-beta-VAE=True --beta=4 --lr=1e-4 --experiment-name=beta_VAE_bs_64 --gpu-devices 0 1 --load-model-checkpoint=True --batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=10 --is-beta-VAE=True --beta=4 --lr=1e-4 --experiment-name=beta_VAE_bs_64 --gpu-devices 0 1 --load-model-checkpoint=True
--batch-size=256 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=10 --lr=1e-4 --experiment-name=VAE_bs_256 --gpu-devices 0 1 --load-model-checkpoint=True --batch-size=256 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=10 --lr=1e-4 --experiment-name=VAE_bs_256 --gpu-devices 0 1 --load-model-checkpoint=True --load-model-checkpoint=True
--batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=10 --lr=1e-4 --experiment-name=VAE_bs_64 --gpu-devices 0 1 --load-model-checkpoint=True --batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=10 --lr=1e-4 --experiment-name=VAE_bs_64 --gpu-devices 0 1 --load-model-checkpoint=True --load-model-checkpoint=True
--batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=15 --is-beta-VAE=True --beta=4 --lr=1e-4 --gpu-devices 0 1 --experiment-name=beta_VAE_bs_64_ls_15 --batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=15 --is-beta-VAE=True --beta=4 --lr=1e-4 --gpu-devices 0 1 --experiment-name=beta_VAE_bs_64_ls_15 --load-model-checkpoint=True
--batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=20 --is-beta-VAE=True --beta=4 --lr=1e-4 --gpu-devices 0 1 --experiment-name=beta_VAE_bs_64_ls_20 --batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=20 --is-beta-VAE=True --beta=4 --lr=1e-4 --gpu-devices 0 1 --experiment-name=beta_VAE_bs_64_ls_20 --load-model-checkpoint=True
--batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=5 --is-beta-VAE=True --beta=4 --lr=1e-4 --gpu-devices 0 1 --experiment-name=beta_VAE_bs_64_ls_5 --batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=5 --is-beta-VAE=True --beta=4 --lr=1e-4 --gpu-devices 0 1 --experiment-name=beta_VAE_bs_64_ls_5 --load-model-checkpoint=True
--batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=5 --lr=1e-4 --gpu-devices 0 1 --experiment-name=VAE_bs_64_ls_5 --batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=5 --lr=1e-4 --gpu-devices 0 1 --experiment-name=VAE_bs_64_ls_5 --load-model-checkpoint=True
--batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=15 --lr=1e-4 --gpu-devices 0 1 --experiment-name=VAE_bs_64_ls_15 --batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=15 --lr=1e-4 --gpu-devices 0 1 --experiment-name=VAE_bs_64_ls_15 --load-model-checkpoint=True
--batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=20 --lr=1e-4 --gpu-devices 0 1 --experiment-name=VAE_bs_64_ls_20 --batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=20 --lr=1e-4 --gpu-devices 0 1 --experiment-name=VAE_bs_64_ls_20 --load-model-checkpoint=True
--batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=10 --lr=5e-4 --gpu-devices 0 1 --experiment-name=VAE_bs_64_ls_10_lr_5e_4 --batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=10 --lr=5e-4 --gpu-devices 0 1 --experiment-name=VAE_bs_64_ls_10_lr_5e_4 --load-model-checkpoint=True
--batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=10 --lr=1e-3 --gpu-devices 0 1 --experiment-name=VAE_bs_64_ls_10_lr_1e_3 --batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=10 --lr=1e-3 --gpu-devices 0 1 --experiment-name=VAE_bs_64_ls_10_lr_1e_3 --load-model-checkpoint=True
\ No newline at end of file
reconstruction_im/charis_VAE_bs_64_ls_15.png

120 KiB

reconstruction_im/charis_VAE_bs_64_ls_20.png

114 KiB

reconstruction_im/charis_VAE_bs_64_ls_5.png

114 KiB

reconstruction_im/charis_beta_VAE_bs_64_ls_15.png

113 KiB

reconstruction_im/charis_beta_VAE_bs_64_ls_20.png

121 KiB

reconstruction_im/charis_beta_VAE_bs_64_ls_5.png

117 KiB

0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment