Upload folder using huggingface_hub

2d_tokenizer_arxiv_image/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:
+ - 16
+ - 16
+ - 256
+ code_shape:
+ - 16
+ - 16
+ - 8
+ 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-arxiv
+ transforms:
+ type: imagenet256x256
+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: 32
+ epochs: 150
+ save_ckpt_freq: 5
+ total_batch_size: 256
+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/in256-rqvae-16x16x8-arxiv-withaux.yaml
+ result_path: /cpfs01/user/cl424408/rq-vae-transformer-main/trained_models/rq2048_16x16x8_imagenet_arxiv_withauxloss
+ 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/rq2048_16x16x8_imagenet_arxiv_withauxloss/in256-rqvae-16x16x8-arxiv-withaux/29072024_055601

2d_tokenizer_arxiv_image/epoch30_model.pt

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

trained_dnd_transformer_arxiv_image_8layer/XXL/xxl_model.pt

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