New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
more problem. #46
Comments
and |
could not unsqueeze empty tensor etc. |
Your input images batches must be of size Bx3x128x128. |
------------ Options ------------- why is that, the size is as you mentioned |
I see a lot of skipped images in your log, this happens when the image is not in the provided dataset path, check that the dataset path is correct. What it may be happening is that the data loader is not being able to read any image. |
@albertpumarola @jackylee1 I think you should check input image, the openface output of mine is *.bmp. |
Closed for inactivity |
python train.py --data_dir mydata --name experiment_1 --batch_size 25
------------ Options -------------
D_adam_b1: 0.5
D_adam_b2: 0.999
G_adam_b1: 0.5
G_adam_b2: 0.999
aus_file: aus_openface.pkl
batch_size: 25
checkpoints_dir: ./checkpoints
cond_nc: 17
data_dir: mydata
dataset_mode: aus
display_freq_s: 300
do_saturate_mask: False
gpu_ids: [0]
image_size: 128
images_folder: imgs
is_train: True
lambda_D_cond: 4000
lambda_D_gp: 10
lambda_D_prob: 1
lambda_cyc: 10
lambda_mask: 0.1
lambda_mask_smooth: 1e-05
load_epoch: 0
lr_D: 0.0001
lr_G: 0.0001
model: ganimation
n_threads_test: 1
n_threads_train: 4
name: experiment_1
nepochs_decay: 10
nepochs_no_decay: 20
num_iters_validate: 1
poses_g_sigma: 0.06
print_freq_s: 60
save_latest_freq_s: 3600
serial_batches: False
test_ids_file: test_ids.csv
train_G_every_n_iterations: 5
train_ids_file: train_ids.csv
-------------- End ----------------
./checkpoints/experiment_1
Dataset AusDataset was created
Dataset AusDataset was created
#train images = 91
#test images = 91
Network generator_wasserstein_gan was created
Network discriminator_wasserstein_gan was created
Model GANimation was created
Traceback (most recent call last):
File "train.py", line 141, in
Train()
File "train.py", line 27, in init
self._train()
File "train.py", line 40, in _train
self._train_epoch(i_epoch)
File "train.py", line 69, in _train_epoch
self._model.optimize_parameters(keep_data_for_visuals=do_visuals, train_generator=train_generator)
File "/data/GANimation/models/ganimation.py", line 207, in optimize_parameters
loss_D, fake_imgs_masked = self._forward_D()
File "/data/GANimation/models/ganimation.py", line 278, in _forward_D
d_real_img_prob, d_real_img_cond = self._D.forward(self._real_img)
File "/data/GANimation/networks/discriminator_wasserstein_gan.py", line 29, in forward
out_aux = self.conv2(h)
File "/usr/local/lib/python2.7/dist-packages/torch/nn/modules/module.py", line 477, in call
Traceback (most recent call last):
File "/usr/lib/python2.7/multiprocessing/util.py", line 274, in _run_finalizers
result = self.forward(*input, **kwargs)
File "/usr/local/lib/python2.7/dist-packages/torch/nn/modules/conv.py", line 301, in forward
finalizer()
File "/usr/lib/python2.7/multiprocessing/util.py", line 207, in call
res = self._callback(*self._args, **self._kwargs)
File "/usr/lib/python2.7/shutil.py", line 252, in rmtree
self.padding, self.dilation, self.groups)
RuntimeError: sizes must be non-negative
The text was updated successfully, but these errors were encountered: