| { |
| "exp_name": "exp_multi_gpu_v1", |
| "model": "prnet", |
| "emb_nn": "dgcnn", |
| "attention": "transformer", |
| "head": "svd", |
| "n_emb_dims": 512, |
| "n_blocks": 1, |
| "n_heads": 4, |
| "n_iters": 3, |
| "discount_factor": 0.9, |
| "n_ff_dims": 1024, |
| "n_keypoints": 512, |
| "temp_factor": 100, |
| "cat_sampler": "softmax", |
| "dropout": 0.0, |
| "batch_size": 16, |
| "test_batch_size": 8, |
| "epochs": 100, |
| "use_sgd": false, |
| "lr": 0.001, |
| "momentum": 0.9, |
| "no_cuda": false, |
| "seed": 1234, |
| "eval": false, |
| "cycle_consistency_loss": 0.1, |
| "feature_alignment_loss": 0.1, |
| "gaussian_noise": false, |
| "unseen": false, |
| "n_points": 1024, |
| "n_subsampled_points": 768, |
| "dataset": "modelnet40", |
| "rot_factor": 4, |
| "model_path": "" |
| } |
