Skip to content

Result with enable bernoulli #20

@mahmoudEltaher

Description

@mahmoudEltaher

Hi, @paschalidoud ,

I try to train the model by Running a command :
python scripts/train_network.py ../../03001627_simple_chair/ tmp/ --epochs 50000 --use_sq --lr 1e-4 --n_primitives 20 --architecture tulsiani --dataset_type shapenet_v2 --use_chamfer --run_on_gpu --batch_size 4

then I test the model
python scripts/forward_pass.py demo/03001627/ tmp/ --model_tag "94ceeee26248a275e8e2378aa23e4253" --n_primitives 20 --weight_file tmp/I77FAV1YI_without_Bernoulli/model_49999 --use_sq --dataset_type shapenet_v2

The result is good as following.
Capture

But once I train_with_bernoulli, the result is not good as following:
Capture

and I notice also that the sparsity regulaizer is always zero , so it is not have effect on the training, what is the effected of this regulizer on the training.
I wonder if I miss something when I apply the model with train_with_bernoulli.

waiting for you kindly advice.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions