I try this cods
CUDA_VISIBLE_DEVICES=1 python train.py --experiment_dir=experiment
--experiment_id=0
--batch_size=32
--lr=0.001
--epoch=30
--sample_steps=100
--schedule=10
--L1_penalty=100
--Lconst_penalty=15
=> I can confirm that my model was trained correctly through the sample picture
CUDA_VISIBLE_DEVICES=1 python infer.py --model_dir=/home/zi2zi/experiment/checkpoint/experiment_0_batch_32
--batch_size=32
--source_obj=/home/zi2zi/experiment/data/val.obj
--embedding_ids=1
--save_dir=save_dir/
=> After executing infer.py, the 'save_dir' was filled with only black pictures....
do u know why??