config.json

{
    "loss": "swd",
    "autoencoder": "pointnet",
    "batch_size": 256,    
    "seed":1,
    
    "device": "cuda",
    "num_prev_losses": 5,
    "add_noise": false,
    "noise_adder":"random",
    "mean_noiseadder":0.0,
    "std_noiseadder": 0.03, 
    "train_denoise": false,

    "embedding_size": 256,
    "input_channels": 3,
    "output_channels": 3,
    "normalize": true,
    "num_points": 2048,

    "prim_caps_size": 1024,
    "prim_vec_size": 16,
    "latent_caps_size": 64,
    "latent_vec_size": 64,

    "version":"mean",

    "max_sw_num_iters": 1,
    "max_sw_lr": 1e-3, 
    
    "num_projs": 100,
    "g_type": "circular",
    "degree": 2, 

    "max_slices": 500,
    "init_projs": 2,
    
    "fix_epsilon": true, 
    "init_rec_epsilon": 0.5,
    "next_epsilon_ratio_rec": 0.01,

    "step_projs": 1,
    "loop_rate_thresh": 0.00001,

    "train_set": "shapenetcore55",
    "train_root": "dataset/shapenet_core55/shapenet57448xyzonly.npz",
    "phase": "train",
    "eval_criteria":"",
    "cates": "",

    "have_val_set": false,
    "val_set": "shapenetcore55",
    "val_root": "dataset/shapenet_core55/shapenet57448xyzonly.npz",
    "val_batch_size": 338,

    "optimizer": "sgd",
    "learning_rate": 0.001,
    "momentum": 0.9,
    "weight_decay": 0.0005,

    "checkpoint": "latest.pth",

    "num_workers": 0,

    "use_scheduler": false,
    "scheduler": "cyclic_lr",
    "base_lr": 0.0001,
    "max_lr": 0.01,

    "evaluator":"based_on_train_loss",

    "best_eval_value": 1e10,
    "best_epoch": -1,

    "best_train_loss": 1e10,
    "best_epoch_based_on_train_loss": -1,

    "epoch_gap_for_evaluation": 1,

    "start_epoch": 0,
    "num_epochs": 301,

    "epoch_gap_for_save": 50,

    "empty_cache_batch": false,

    "empty_cache_epoch": false
}