UnCRtainTS/model/inference/monotemporalL2/conf.json

{
    "model": "uncrtaints",
    "encoder_widths": [
        128
    ],
    "decoder_widths": [
        128,
        128,
        128,
        128,
        128
    ],
    "out_conv": [
        13
    ],
    "mean_nonLinearity": true,
    "var_nonLinearity": "softplus",
    "use_sar": true,
    "agg_mode": "att_group",
    "encoder_norm": "group",
    "decoder_norm": "batch",
    "n_head": 1,
    "d_model": 256,
    "use_v": false,
    "positional_encoding": true,
    "d_k": 4,
    "experiment_name": "monotemporalL2",
    "lr": 0.001,
    "gamma": 0.8,
    "ref_date": "2014-04-03",
    "pad_value": 0,
    "padding_mode": "reflect",
    "val_every": 1,
    "val_after": 0,
    "pretrain": true,
    "sample_type": "pretrain",
    "vary_samples": true,
    "max_samples": 1000000000,
    "input_size": 256,
    "loss": "l2",
    "covmode": "diag",
    "scale_by": 10.0,
    "separate_out": false,
    "resume_from": false,
    "epochs": 20,
    "res_dir": "./inference",
    "plot_every": -1,
    "export_every": 1,
    "resume_at": -1,
    "device": "cuda:6",
    "display_step": 10,
    "batch_size": 128,
    "input_t": 1,
    "min_cov": 0.0,
    "max_cov": 1.0,
    "root1": "/home/data/SEN12MSCRTS",
    "root2": "/home/data/SEN12MSCRTS",
    "root3": "data2/SEN12MSCR",
    "region": "all",
    "max_samples_count": 1000000000,
    "trained_checkp": "",
    "weight_folder": "checkpoint/",
    "pid": 2973877,
    "block_type": "mbconv",
    "low_res_size": 32,
    "num_workers": 0,
    "rdm_seed": 1,
    "unfreeze_after": 0,
    "chunk_size": null,
    "precomputed": "/home/code/UnCRtainTS/util/precomputed",
    "max_samples_frac": 1.0,
    "profile": false,
    "use_custom": false,
    "load_config": ""
}