update loss plot 2

VAE_model/models.py

@@ -40,8 +40,8 @@ class VAE(nn.Module):
         self.stride = stride
         self.num_pixels = img_size[1] * img_size[2]
         self.temperature = temperature
-        self.hidden_dim = 256  # Hidden dimension of linear layer
-        self.reshape = (256, 1, 1)  # Shape required to start transpose convs
+        self.hidden_dim = nb_filter_conv4*4  # Hidden dimension of linear layer
+        self.reshape = (self.hidden_dim, 1, 1)  # Shape required to start transpose convs
 
         # Calculate dimensions of latent distribution
         self.latent_cont_dim = 0

parameters_combinations/param_combinations_chairs_test.txt

---batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=10 --lr=1e-4 --experiment-name=VAE_bs_64_conv_64_64_128_128 --gpu-devices 0 1 --nb-filter-conv1_64 --nb-filter-conv2=64 --nb-filter-conv3=128 --nb-filter-conv4=128
---batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=10 --lr=1e-4 --experiment-name=VAE_bs_64_conv_128_128_256_256 --gpu-devices 0 1 --nb-filter-conv1_128 --nb-filter-conv2=128 --nb-filter-conv3=256 --nb-filter-conv4=256
---batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=30 --lr=1e-4 --experiment-name=VAE_bs_64_conv_64_64_128_128_ls_30 --gpu-devices 0 1 --nb-filter-conv1_64 --nb-filter-conv2=64 --nb-filter-conv3=128 --nb-filter-conv4=128
---batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=30 --lr=1e-4 --experiment-name=VAE_bs_64_conv_128_128_256_256_ls_30 --gpu-devices 0 1 --nb-filter-conv1_128 --nb-filter-conv2=128 --nb-filter-conv3=256 --nb-filter-conv4=256
+--batch-size=64 --dataset=rendered_chairs --epochs=4 --latent_spec_cont=10 --lr=1e-4 --experiment-name=VAE_bs_64_conv_64_64_128_128 --gpu-devices 0 1 --nb-filter-conv1=64 --nb-filter-conv2=64 --nb-filter-conv3=128 --nb-filter-conv4=128
+--batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=10 --lr=1e-4 --experiment-name=VAE_bs_64_conv_128_128_256_256 --gpu-devices 0 1 --nb-filter-conv1=128 --nb-filter-conv2=128 --nb-filter-conv3=256 --nb-filter-conv4=256
+--batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=30 --lr=1e-4 --experiment-name=VAE_bs_64_conv_64_64_128_128_ls_30 --gpu-devices 0 1 --nb-filter-conv1=64 --nb-filter-conv2=64 --nb-filter-conv3=128 --nb-filter-conv4=128
+--batch-size=64 --dataset=rendered_chairs --epochs=400 --latent_spec_cont=30 --lr=1e-4 --experiment-name=VAE_bs_64_conv_128_128_256_256_ls_30 --gpu-devices 0 1 --nb-filter-conv1=128 --nb-filter-conv2=128 --nb-filter-conv3=256 --nb-filter-conv4=256