NimaBoscarino's picture
include the model in the Space
ee454f2
raw history blame
No virus
6.73 kB
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