-
Notifications
You must be signed in to change notification settings - Fork 31
Description
------------ Options -------------
batchSize: 4
beta1: 0.5
checkpoints_dir: ./checkpoints
continue_train: False
cross_embed: False
data_augmentation: True
data_type: 32
dataroot: ./datasets/helen_align/
debug: False
debug_mask_part: True
display_freq: 30
display_winsize: 512
dist_model: False
embed_length: 128
embed_nc: 64
face_rec_model:
feat_num: 3
fineSize: 512
gpu_ids: [0, 1]
input_nc: 3
instance_feat: False
isTrain: True
label_feat: False
label_nc: 11
lambda_feat: 10.0
loadSize: 1024
load_features: False
load_pretrain:
longSize: 256
lr: 0.0002
mask_model: True
max_dataset_size: inf
model: pix2pixHD
nThreads: 2
n_blocks_global: 9
n_blocks_local: 3
n_clusters: 10
n_downsample_E: 4
n_downsample_global: 2
n_layers_D: 3
n_local_enhancers: 1
name: face_256
ndf: 64
nef: 16
netG: global
ngf: 64
niter: 10
niter_decay: 45
niter_fix_global: 0
no_flip: False
no_ganFeat_loss: False
no_grammatrix_loss: False
no_html: False
no_instance: True
no_l2_loss: False
no_lsgan: True
no_vgg_loss: False
norm: batch
num_D: 2
output_nc: 3
phase: train
pool_size: 0
print_freq: 100
resize_or_crop: none
sample_noise: False
save_epoch_freq: 10
save_latest_freq: 15000
serial_batches: False
tf_log: False
use_dropout: False
use_parsing_label: False
vae_encoder: True
verbose: True
which_epoch: latest
-------------- End ----------------
CustomDatasetDataLoader
dataset [AlignedDataset] was created
Traceback (most recent call last):
File "train.py", line 39, in
data_loader = CreateDataLoader(opt)
File "/content/Mask_Guided_Portrait_Editing/data/data_loader.py", line 6, in CreateDataLoader
data_loader.initialize(opt)
File "/content/Mask_Guided_Portrait_Editing/data/custom_dataset_data_loader.py", line 20, in initialize
self.dataset = CreateDataset(opt)
File "/content/Mask_Guided_Portrait_Editing/data/custom_dataset_data_loader.py", line 11, in CreateDataset
dataset.initialize(opt)
File "/content/Mask_Guided_Portrait_Editing/data/aligned_dataset.py", line 23, in initialize
self.A_paths = sorted(make_dataset(self.dir_A))
File "/content/Mask_Guided_Portrait_Editing/data/image_folder.py", line 23, in make_dataset
assert os.path.isdir(dir), '%s is not a valid directory' % dir
AssertionError: ./datasets/helen_align/train_label is not a valid directory