Upload folder using huggingface_hub

2d_tokenizer_imagenet512/config.yaml

@@ -0,0 +1,126 @@
+arch:
+  ema: null
+  hparams:
+    loss_type: mse
+    restart_unused_codes: true
+    use_padding_idx: false
+    masked_dropout: 0.0
+    bottleneck_type: rq
+    embed_dim: 256
+    n_embed: 16384
+    latent_shape:
+    - 32
+    - 32
+    - 256
+    code_shape:
+    - 32
+    - 32
+    - 2
+    shared_codebook: true
+    decay: 0.99
+    latent_loss_weight: 0.25
+  checkpointing: true
+  type: rq-vae
+  code_hier: 1
+  ddconfig:
+    double_z: false
+    z_channels: 256
+    resolution: 256
+    in_channels: 3
+    out_ch: 3
+    ch: 128
+    ch_mult:
+    - 1
+    - 1
+    - 2
+    - 2
+    - 4
+    num_res_blocks: 2
+    attn_resolutions:
+    - 8
+    dropout: 0.0
+dataset:
+  transform:
+    type: null
+  type: imagenet
+  transforms:
+    type: imagenet512x512
+optimizer:
+  type: adam
+  max_gn: null
+  warmup:
+    mode: fix
+    start_from_zero: true
+    epoch: 0.5
+    multiplier: 1
+    buffer_epoch: 0
+    min_lr: 4.0e-05
+  init_lr: 4.0e-05
+  weight_decay: 0.0
+  betas:
+  - 0.5
+  - 0.9
+  grad_accm_steps: 1
+experiment:
+  test_freq: 1
+  amp: false
+  batch_size: 20
+  epochs: 150
+  save_ckpt_freq: 5
+  total_batch_size: 160
+gan:
+  disc:
+    optimizer:
+      type: adam
+      max_gn: null
+      warmup:
+        mode: fix
+        start_from_zero: true
+        epoch: 0.5
+        multiplier: 1
+        buffer_epoch: 0
+        min_lr: 4.0e-05
+      init_lr: 4.0e-05
+      weight_decay: 0.0
+      betas:
+      - 0.5
+      - 0.9
+    arch:
+      in_channels: 3
+      num_layers: 2
+      use_actnorm: false
+      ndf: 64
+      spectral_norm: false
+  loss:
+    disc_loss: hinge
+    gen_loss: vanilla
+    disc_weight: 0.75
+    perceptual_weight: 1.0
+    disc_start: 0
+seed: 0
+runtime:
+  args:
+    model_config: /cpfs01/user/cl424408/rq-vae-transformer-main/configs/imagenet256/stage1/in512-rqvae-32x32x2-arxiv-withaux.yaml
+    result_path: /cpfs01/user/cl424408/rq-vae-transformer-main/trained_models/in512_32x32x2_imagenet_shared_bsz20*8
+    load_path: ''
+    test_batch_size: 200
+    test_epoch: -1
+    postfix: ''
+    seed: 0
+    world_size: 8
+    local_rank: 0
+    node_rank: -1
+    dist_backend: nccl
+    timeout: 86400
+    eval: false
+    resume: false
+    rank: 0
+  extra_config: {}
+  distenv:
+    world_size: 8
+    world_rank: 0
+    local_rank: 0
+    num_gpus: 1
+    master: true
+    device_name: NVIDIA A100-SXM4-80GB
+result_path: /cpfs01/user/cl424408/rq-vae-transformer-main/trained_models/in512_32x32x2_imagenet_shared_bsz20*8/in512-rqvae-32x32x2-arxiv-withaux/17092024_101256

2d_tokenizer_imagenet512/model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ebc62c9a5a87cd31039898d2290c35b87dd83ceacd1cd3b540164f0132455ada
+size 836123089