donydchen / ganimation_replicate

An Out-of-the-Box Replication of GANimation using PyTorch, pretrained weights are available!
MIT License
247 stars 48 forks source link

_pickle.UnpicklingError: pickle data was truncated #24

Closed GuangyuanLiu1999 closed 2 years ago

GuangyuanLiu1999 commented 2 years ago

python main.py --mode test --data_root datasets/celebA/ --batch_size 8 --max_dataset_size 150 --gpu_ids 0,1 --ckpt_dir ckpts/celebA/ganimation/190327_161852/ --load_epoch 30 --save_test_gif ------------------- [ test][220228_200812]Options -------------------- aus_nc: 17
aus_pkl: aus_openface.pkl
batch_size: 8 [default: 25] beta1: 0.5
ckpt_dir: ckpts/celebA/ganimation/190327_161852/ [default: ./ckpts] data_root: datasets/celebA/ [default: None] epoch_count: 1
final_size: 128
gan_type: wgan-gp
gpu_ids: [0, 1] [default: 0] img_nc: 3
imgs_dir: imgs
init_gain: 0.02
init_type: normal
interpolate_len: 5
lambda_aus: 160.0
lambda_dis: 1.0
lambda_mask: 0
lambda_rec: 10.0
lambda_tv: 0
lambda_wgan_gp: 10.0
load_epoch: 30 [default: 0] load_size: 148
log_file: logs.txt
lr: 0.0001
lr_decay_iters: 50
lr_policy: lambda
lucky_seed: 1646050092 [default: 0] max_dataset_size: 150 [default: inf] mode: test [default: train] model: ganimation
n_threads: 6
name: 220228_200812
ndf: 64
ngf: 64
niter: 20
niter_decay: 10
no_aus_noise: False
no_flip: False
no_test_eval: False
norm: instance
opt_file: opt.txt
plot_losses_freq: 20000
print_losses_freq: 100
resize_or_crop: none
results: results/celebA_ganimation_30 [default: results] sample_img_freq: 2000
save_epoch_freq: 2
save_test_gif: True [default: False] serial_batches: False
test_csv: test_ids.csv
train_csv: train_ids.csv
train_gen_iter: 5
use_dropout: False
visdom_display_id: 0 [default: 1] visdom_env: main
visdom_port: 8097
--------------------- [ test][220228_200812]End ----------------------

Traceback (most recent call last): File "main.py", line 16, in solver.run_solver() File "/home/scs4450/lgy/GANimation/solvers.py", line 38, in run_solver self.test_networks(self.opt) File "/home/scs4450/lgy/GANimation/solvers.py", line 105, in test_networks self.init_test_setting(opt) File "/home/scs4450/lgy/GANimation/solvers.py", line 109, in init_test_setting self.test_dataset = create_dataloader(opt) File "/home/scs4450/lgy/GANimation/data/data_loader.py", line 15, in create_dataloader data_loader.initialize(opt) File "/home/scs4450/lgy/GANimation/data/data_loader.py", line 35, in initialize self.dataset = self.create_datase() File "/home/scs4450/lgy/GANimation/data/data_loader.py", line 30, in create_datase dataset.initialize(self.opt) File "/home/scs4450/lgy/GANimation/data/celeba.py", line 13, in initialize super(CelebADataset, self).initialize(opt) File "/home/scs4450/lgy/GANimation/data/base_dataset.py", line 31, in initialize self.aus_dict = self.load_dict(aus_pkl) File "/home/scs4450/lgy/GANimation/data/base_dataset.py", line 42, in load_dict saved_dict = pickle.load(f, encoding='latin1') _pickle.UnpicklingError: pickle data was truncated

Excuse me, does this error mean that the loaded pretrained model file is corrupt?