Spaces:
Runtime error
Runtime error
comet: | |
display_size: 20 | |
im_per_row: | |
d: 4 | |
m: 6 | |
p: 4 | |
s: 4 | |
rows_per_log: 5 | |
tags: [] | |
data: | |
check_samples: false | |
files: | |
base: /miniscratch/_groups/ccai/data/jsons | |
train: | |
kitti: /miniscratch/_groups/ccai/data/jsons/train_kitti.json | |
r: /miniscratch/_groups/ccai/data/jsons/train_r_full.json | |
rf: /miniscratch/_groups/ccai/data/jsons/train_rf.json | |
s: /miniscratch/_groups/ccai/data/jsons/train_s_fixedholes.json | |
val: | |
kitti: /miniscratch/_groups/ccai/data/jsons/val_kitti.json | |
r: /miniscratch/_groups/ccai/data/jsons/val_r_full.json | |
rf: /miniscratch/_groups/ccai/data/jsons/val_rf_labelbox.json | |
s: /miniscratch/_groups/ccai/data/jsons/val_s_fixedholes.json | |
loaders: | |
batch_size: 4 | |
num_workers: 4 | |
max_samples: -1 | |
normalization: default | |
transforms: | |
- ignore: val | |
name: hflip | |
p: 0.5 | |
- ignore: false | |
keep_aspect_ratio: true | |
name: resize | |
new_size: 640 | |
- center: val | |
height: 600 | |
ignore: false | |
name: crop | |
width: 600 | |
- ignore: val | |
name: brightness | |
- ignore: val | |
name: saturation | |
- ignore: val | |
name: contrast | |
- ignore: false | |
name: resize | |
new_size: | |
d: 160 | |
default: 640 | |
s: 160 | |
dis: | |
default: | |
get_intermediate_features: false | |
init_gain: 0.02 | |
init_type: xavier | |
input_nc: 3 | |
n_layers: 4 | |
ndf: 64 | |
norm: instance | |
num_D: 1 | |
use_sigmoid: false | |
flip_prob: 0.05 | |
m: | |
architecture: base | |
gan_type: WGAN_norm | |
get_intermediate_features: false | |
init_gain: 0.02 | |
init_type: xavier | |
input_nc: 3 | |
multi_level: false | |
n_layers: 4 | |
ndf: 64 | |
norm: instance | |
num_D: 1 | |
use_sigmoid: false | |
wgan_clamp_lower: -0.01 | |
wgan_clamp_upper: 0.01 | |
opt: | |
beta1: 0.5 | |
lr: | |
default: 2.0e-05 | |
lr_gamma: 0.5 | |
lr_milestones: 5 | |
lr_policy: step | |
lr_step_size: 15 | |
optimizer: ExtraAdam | |
p: | |
get_intermediate_features: true | |
init_gain: 0.02 | |
init_type: xavier | |
input_nc: 3 | |
n_layers: 4 | |
ndf: 64 | |
norm: instance | |
num_D: 3 | |
use_local_discriminator: false | |
use_sigmoid: false | |
s: | |
gan_type: WGAN_norm | |
get_intermediate_features: false | |
init_gain: 0.02 | |
init_type: xavier | |
input_nc: 3 | |
n_layers: 4 | |
ndf: 64 | |
norm: instance | |
num_D: 1 | |
use_sigmoid: false | |
wgan_clamp_lower: -0.01 | |
wgan_clamp_upper: 0.01 | |
soft_shift: 0.2 | |
domains: | |
- r | |
- s | |
gen: | |
d: | |
activ: lrelu | |
architecture: dada | |
classify: | |
enable: false | |
linspace: | |
buckets: 256 | |
max: 6.95 | |
min: 0.35 | |
init_gain: 0.02 | |
init_type: xavier | |
loss: sigm | |
n_downsample: 3 | |
n_res: 1 | |
n_upsample: 3 | |
norm: batch | |
output_dim: 1 | |
pad_type: reflect | |
proj_dim: 32 | |
upsample_featuremaps: true | |
deeplabv2: | |
nblocks: | |
- 3 | |
- 4 | |
- 23 | |
- 3 | |
pretrained_model: /miniscratch/_groups/ccai/data/pretrained_models/deeplabv2/DeepLab_resnet_pretrained_imagenet.pth | |
use_pretrained: true | |
deeplabv3: | |
backbone: resnet | |
output_stride: 8 | |
pretrained_model: | |
mobilenet: /miniscratch/_groups/ccai/data/pretrained_models/deeplabv3/deeplabv3_plus_mobilenetv2_segmentron.pth | |
resnet: /miniscratch/_groups/ccai/data/pretrained_models/deeplabv3/model_CoinCheungDeepLab-v3-plus.pth | |
use_pretrained: true | |
default: | |
activ: lrelu | |
init_gain: 0.02 | |
init_type: xavier | |
n_downsample: 3 | |
n_res: 1 | |
n_upsample: 3 | |
norm: spectral | |
pad_type: reflect | |
proj_dim: 32 | |
encoder: | |
activ: lrelu | |
architecture: deeplabv3 | |
dim: 32 | |
init_gain: 0.02 | |
init_type: xavier | |
input_dim: 3 | |
n_downsample: 3 | |
n_res: 0 | |
n_upsample: 3 | |
norm: spectral | |
pad_type: reflect | |
proj_dim: 32 | |
m: | |
activ: lrelu | |
init_gain: 0.02 | |
init_type: xavier | |
n_downsample: 3 | |
n_res: 3 | |
n_upsample: 3 | |
norm: spectral | |
output_dim: 1 | |
pad_type: reflect | |
proj_dim: 64 | |
spade: | |
activations: | |
all_lrelu: true | |
cond_nc: 15 | |
detach: false | |
latent_dim: 128 | |
num_layers: 3 | |
spade_param_free_norm: batch | |
spade_use_spectral_norm: true | |
use_advent: true | |
use_ground_intersection: true | |
use_low_level_feats: true | |
use_minent: true | |
use_minent_var: true | |
use_pl4m: false | |
use_proj: true | |
use_spade: true | |
opt: | |
beta1: 0.9 | |
lr: | |
default: 5.0e-05 | |
lr_gamma: 0.5 | |
lr_milestones: 15 | |
lr_policy: step | |
lr_step_size: 5 | |
optimizer: ExtraAdam | |
p: | |
activ: lrelu | |
diff_aug: | |
cutout_ratio: 0.5 | |
do_color_jittering: false | |
do_cutout: false | |
do_translation: false | |
translation_ratio: 0.125 | |
use: false | |
init_gain: 0.02 | |
init_type: xavier | |
latent_dim: 640 | |
loss: gan | |
n_downsample: 3 | |
n_res: 1 | |
n_upsample: 3 | |
no_z: true | |
norm: spectral | |
output_dim: 3 | |
pad_type: reflect | |
paste_original_content: true | |
pl4m_epoch: 49 | |
proj_dim: 32 | |
spade_kernel_size: 3 | |
spade_n_up: 7 | |
spade_param_free_norm: instance | |
spade_use_spectral_norm: true | |
use_final_shortcut: false | |
s: | |
activ: lrelu | |
architecture: deeplabv3 | |
depth_dada_fusion: true | |
depth_feat_fusion: true | |
init_gain: 0.02 | |
init_type: xavier | |
n_downsample: 3 | |
n_res: 1 | |
n_upsample: 3 | |
norm: spectral | |
num_classes: 11 | |
output_dim: 11 | |
pad_type: reflect | |
proj_dim: 32 | |
upsample_featuremaps: false | |
use_advent: true | |
use_minent: true | |
load_paths: | |
m: config/model/masker/checkpoints/latest_ckpt.pth # relative to $PWD, *not* this file | |
p: none | |
pm: none | |
output_path: /miniscratch/_groups/ccai/experiments/runs/ablation-v0/out--49167587 | |
tasks: | |
- m | |
- s | |
- d | |
train: | |
amp: false | |
auto_resume: true | |
epochs: 50 | |
fid: | |
batch_size: 50 | |
dims: 2048 | |
n_images: 57 | |
kitti: | |
batch_size: 6 | |
epochs: 10 | |
pretrain: false | |
lambdas: | |
C: 1 | |
G: | |
classifier: 1 | |
d: | |
gml: 0.5 | |
main: 1 | |
m: | |
bce: 1 | |
gi: 0.05 | |
pl4m: 1 | |
tv: 1 | |
p: | |
context: 0 | |
dm: 1 | |
featmatch: 10 | |
gan: 1 | |
reconstruction: 0 | |
tv: 0 | |
vgg: 10 | |
s: | |
advent: 0.001 | |
crossent: 1 | |
crossent_pseudo: 0.001 | |
minent: 0.001 | |
advent: | |
WGAN_gp: 10 | |
adv_aux: 0.0 | |
adv_main: 1.0 | |
dis_aux: 0.0 | |
dis_main: 1.0 | |
ent_aux: 0.0 | |
ent_main: 0.5 | |
ent_var: 0.1 | |
latent_domain_adaptation: false | |
log_level: 2 | |
min_save_epoch: 28 | |
pseudo: | |
epochs: 10 | |
tasks: | |
- m | |
- s | |
- d | |
resume: false | |
save_n_epochs: 25 | |
val: | |
store_images: false | |
val_painter: config/model/painter/checkpoints/latest_ckpt.pth # relative to $PWD, *not* this file | |