├── .gitignore ├── .project-root ├── License.txt ├── README.md ├── assets ├── comp.jpg ├── method.jpg ├── recon │ ├── beer_gt.gif │ ├── beer_recon.gif │ ├── camel_gt.gif │ ├── camel_recon.gif │ ├── car_gt.gif │ ├── car_recon.gif │ ├── george_gt.gif │ └── george_recon.gif ├── showcase │ ├── balloon.gif │ ├── car.gif │ ├── city.gif │ ├── dog.gif │ ├── firework.gif │ ├── lake.gif │ ├── make_up.gif │ ├── smile.gif │ ├── walk.gif │ ├── water.gif │ ├── waterfall.gif │ └── wave.gif └── story │ ├── dentist1.gif │ ├── dentist2.gif │ ├── dentist3.gif │ ├── kitchen1.gif │ ├── kitchen2.gif │ └── kitchen3.gif ├── configs ├── accelerate │ ├── deepspeed_stage_2_offload.yaml │ └── stage2_offload_bf16_dp.json ├── clm_models │ ├── agent_7b_in64_out64_video_gmm.yaml │ ├── agent_7b_in64_out64_video_gmm_pretrain.yaml │ ├── agent_7b_in64_out64_video_gmm_pretrain_comp.yaml │ ├── agent_7b_in64_out64_video_gmm_pretrain_gen.yaml │ ├── agent_7b_in64_out64_video_gmm_sft_comp.yaml │ ├── agent_7b_in64_out64_video_gmm_sft_gen.yaml │ ├── mistral7b_lora.yaml │ ├── mistral7b_lora_pretrain.yaml │ ├── mistral7b_merged_pretrain.yaml │ ├── mistral7b_merged_sft_comp.yaml │ └── mistral7b_merged_sft_gen.yaml ├── data │ ├── clm_pretrain_multi_data_video_comp_gen.yaml │ ├── clm_sft_multi_data_video_comp.yaml │ └── clm_sft_multi_data_video_gen.yaml ├── processer │ ├── dc_256_transform.yaml │ └── dc_256_transform_fixed.yaml ├── tokenizer │ └── Divot_mistral_tokenizer_instruct.yaml ├── video_detokenizer │ ├── Divot_detokenizer_stage1.yaml │ └── Divot_detokenizer_stage2.yaml └── visual_encoder │ └── Divot_video.yaml ├── demo_videos ├── car.mp4 └── wine.mp4 ├── lvdm ├── basics.py ├── common.py ├── distributions.py ├── ema.py ├── models │ ├── autoencoder.py │ ├── samplers │ │ ├── ddim.py │ │ └── ddim_multiplecond.py │ └── utils_diffusion.py └── modules │ ├── attention.py │ ├── networks │ ├── ae_modules.py │ └── openaimodel3d.py │ └── x_transformer.py ├── requirements.txt ├── scripts ├── train_Divot_pretrain_comp_gen.sh ├── train_Divot_sft_comp.sh └── train_Divot_sft_gen.sh └── src ├── data ├── __init__.py ├── control_video.py ├── dataloader_utils.py └── datapipes.py ├── models ├── dc_detokenizer.py ├── dc_encoder.py ├── mistral │ ├── configuration_mistral.py │ ├── convert_mistral_weights_to_hf.py │ ├── modeling_attn_mask_utils.py │ └── modeling_mistral.py └── vit.py ├── models_clm ├── __init__.py ├── generation.py ├── models.py └── peft_models.py ├── processer ├── tokenizer.py └── transforms.py ├── tools ├── eval_Divot_video_comp.py ├── eval_Divot_video_gen.py ├── eval_Divot_video_recon.py └── merge_agent_lora_weight.py └── train ├── dist_utils.py ├── optim.py ├── schedular.py └── train_Divot.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/.gitignore -------------------------------------------------------------------------------- /.project-root: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /License.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/License.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/README.md -------------------------------------------------------------------------------- /assets/comp.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/comp.jpg -------------------------------------------------------------------------------- /assets/method.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/method.jpg -------------------------------------------------------------------------------- /assets/recon/beer_gt.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/recon/beer_gt.gif -------------------------------------------------------------------------------- /assets/recon/beer_recon.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/recon/beer_recon.gif -------------------------------------------------------------------------------- /assets/recon/camel_gt.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/recon/camel_gt.gif -------------------------------------------------------------------------------- /assets/recon/camel_recon.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/recon/camel_recon.gif -------------------------------------------------------------------------------- /assets/recon/car_gt.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/recon/car_gt.gif -------------------------------------------------------------------------------- /assets/recon/car_recon.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/recon/car_recon.gif -------------------------------------------------------------------------------- /assets/recon/george_gt.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/recon/george_gt.gif -------------------------------------------------------------------------------- /assets/recon/george_recon.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/recon/george_recon.gif -------------------------------------------------------------------------------- /assets/showcase/balloon.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/showcase/balloon.gif -------------------------------------------------------------------------------- /assets/showcase/car.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/showcase/car.gif -------------------------------------------------------------------------------- /assets/showcase/city.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/showcase/city.gif -------------------------------------------------------------------------------- /assets/showcase/dog.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/showcase/dog.gif -------------------------------------------------------------------------------- /assets/showcase/firework.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/showcase/firework.gif -------------------------------------------------------------------------------- /assets/showcase/lake.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/showcase/lake.gif -------------------------------------------------------------------------------- /assets/showcase/make_up.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/showcase/make_up.gif -------------------------------------------------------------------------------- /assets/showcase/smile.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/showcase/smile.gif -------------------------------------------------------------------------------- /assets/showcase/walk.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/showcase/walk.gif -------------------------------------------------------------------------------- /assets/showcase/water.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/showcase/water.gif -------------------------------------------------------------------------------- /assets/showcase/waterfall.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/showcase/waterfall.gif -------------------------------------------------------------------------------- /assets/showcase/wave.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/showcase/wave.gif -------------------------------------------------------------------------------- /assets/story/dentist1.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/story/dentist1.gif -------------------------------------------------------------------------------- /assets/story/dentist2.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/story/dentist2.gif -------------------------------------------------------------------------------- /assets/story/dentist3.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/story/dentist3.gif -------------------------------------------------------------------------------- /assets/story/kitchen1.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/story/kitchen1.gif -------------------------------------------------------------------------------- /assets/story/kitchen2.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/story/kitchen2.gif -------------------------------------------------------------------------------- /assets/story/kitchen3.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/assets/story/kitchen3.gif -------------------------------------------------------------------------------- /configs/accelerate/deepspeed_stage_2_offload.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/accelerate/deepspeed_stage_2_offload.yaml -------------------------------------------------------------------------------- /configs/accelerate/stage2_offload_bf16_dp.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/accelerate/stage2_offload_bf16_dp.json -------------------------------------------------------------------------------- /configs/clm_models/agent_7b_in64_out64_video_gmm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/clm_models/agent_7b_in64_out64_video_gmm.yaml -------------------------------------------------------------------------------- /configs/clm_models/agent_7b_in64_out64_video_gmm_pretrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/clm_models/agent_7b_in64_out64_video_gmm_pretrain.yaml -------------------------------------------------------------------------------- /configs/clm_models/agent_7b_in64_out64_video_gmm_pretrain_comp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/clm_models/agent_7b_in64_out64_video_gmm_pretrain_comp.yaml -------------------------------------------------------------------------------- /configs/clm_models/agent_7b_in64_out64_video_gmm_pretrain_gen.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/clm_models/agent_7b_in64_out64_video_gmm_pretrain_gen.yaml -------------------------------------------------------------------------------- /configs/clm_models/agent_7b_in64_out64_video_gmm_sft_comp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/clm_models/agent_7b_in64_out64_video_gmm_sft_comp.yaml -------------------------------------------------------------------------------- /configs/clm_models/agent_7b_in64_out64_video_gmm_sft_gen.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/clm_models/agent_7b_in64_out64_video_gmm_sft_gen.yaml -------------------------------------------------------------------------------- /configs/clm_models/mistral7b_lora.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/clm_models/mistral7b_lora.yaml -------------------------------------------------------------------------------- /configs/clm_models/mistral7b_lora_pretrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/clm_models/mistral7b_lora_pretrain.yaml -------------------------------------------------------------------------------- /configs/clm_models/mistral7b_merged_pretrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/clm_models/mistral7b_merged_pretrain.yaml -------------------------------------------------------------------------------- /configs/clm_models/mistral7b_merged_sft_comp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/clm_models/mistral7b_merged_sft_comp.yaml -------------------------------------------------------------------------------- /configs/clm_models/mistral7b_merged_sft_gen.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/clm_models/mistral7b_merged_sft_gen.yaml -------------------------------------------------------------------------------- /configs/data/clm_pretrain_multi_data_video_comp_gen.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/data/clm_pretrain_multi_data_video_comp_gen.yaml -------------------------------------------------------------------------------- /configs/data/clm_sft_multi_data_video_comp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/data/clm_sft_multi_data_video_comp.yaml -------------------------------------------------------------------------------- /configs/data/clm_sft_multi_data_video_gen.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/data/clm_sft_multi_data_video_gen.yaml -------------------------------------------------------------------------------- /configs/processer/dc_256_transform.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/processer/dc_256_transform.yaml -------------------------------------------------------------------------------- /configs/processer/dc_256_transform_fixed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/processer/dc_256_transform_fixed.yaml -------------------------------------------------------------------------------- /configs/tokenizer/Divot_mistral_tokenizer_instruct.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/tokenizer/Divot_mistral_tokenizer_instruct.yaml -------------------------------------------------------------------------------- /configs/video_detokenizer/Divot_detokenizer_stage1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/video_detokenizer/Divot_detokenizer_stage1.yaml -------------------------------------------------------------------------------- /configs/video_detokenizer/Divot_detokenizer_stage2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/video_detokenizer/Divot_detokenizer_stage2.yaml -------------------------------------------------------------------------------- /configs/visual_encoder/Divot_video.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/configs/visual_encoder/Divot_video.yaml -------------------------------------------------------------------------------- /demo_videos/car.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/demo_videos/car.mp4 -------------------------------------------------------------------------------- /demo_videos/wine.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/demo_videos/wine.mp4 -------------------------------------------------------------------------------- /lvdm/basics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/lvdm/basics.py -------------------------------------------------------------------------------- /lvdm/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/lvdm/common.py -------------------------------------------------------------------------------- /lvdm/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/lvdm/distributions.py -------------------------------------------------------------------------------- /lvdm/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/lvdm/ema.py -------------------------------------------------------------------------------- /lvdm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/lvdm/models/autoencoder.py -------------------------------------------------------------------------------- /lvdm/models/samplers/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/lvdm/models/samplers/ddim.py -------------------------------------------------------------------------------- /lvdm/models/samplers/ddim_multiplecond.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/lvdm/models/samplers/ddim_multiplecond.py -------------------------------------------------------------------------------- /lvdm/models/utils_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/lvdm/models/utils_diffusion.py -------------------------------------------------------------------------------- /lvdm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/lvdm/modules/attention.py -------------------------------------------------------------------------------- /lvdm/modules/networks/ae_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/lvdm/modules/networks/ae_modules.py -------------------------------------------------------------------------------- /lvdm/modules/networks/openaimodel3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/lvdm/modules/networks/openaimodel3d.py -------------------------------------------------------------------------------- /lvdm/modules/x_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/lvdm/modules/x_transformer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/train_Divot_pretrain_comp_gen.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/scripts/train_Divot_pretrain_comp_gen.sh -------------------------------------------------------------------------------- /scripts/train_Divot_sft_comp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/scripts/train_Divot_sft_comp.sh -------------------------------------------------------------------------------- /scripts/train_Divot_sft_gen.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/scripts/train_Divot_sft_gen.sh -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- 1 | from .datapipes import TarArchiveLoader, JsonlParserIterDataPipe 2 | -------------------------------------------------------------------------------- /src/data/control_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/data/control_video.py -------------------------------------------------------------------------------- /src/data/dataloader_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/data/dataloader_utils.py -------------------------------------------------------------------------------- /src/data/datapipes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/data/datapipes.py -------------------------------------------------------------------------------- /src/models/dc_detokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/models/dc_detokenizer.py -------------------------------------------------------------------------------- /src/models/dc_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/models/dc_encoder.py -------------------------------------------------------------------------------- /src/models/mistral/configuration_mistral.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/models/mistral/configuration_mistral.py -------------------------------------------------------------------------------- /src/models/mistral/convert_mistral_weights_to_hf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/models/mistral/convert_mistral_weights_to_hf.py -------------------------------------------------------------------------------- /src/models/mistral/modeling_attn_mask_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/models/mistral/modeling_attn_mask_utils.py -------------------------------------------------------------------------------- /src/models/mistral/modeling_mistral.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/models/mistral/modeling_mistral.py -------------------------------------------------------------------------------- /src/models/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/models/vit.py -------------------------------------------------------------------------------- /src/models_clm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models_clm/generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/models_clm/generation.py -------------------------------------------------------------------------------- /src/models_clm/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/models_clm/models.py -------------------------------------------------------------------------------- /src/models_clm/peft_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/models_clm/peft_models.py -------------------------------------------------------------------------------- /src/processer/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/processer/tokenizer.py -------------------------------------------------------------------------------- /src/processer/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/processer/transforms.py -------------------------------------------------------------------------------- /src/tools/eval_Divot_video_comp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/tools/eval_Divot_video_comp.py -------------------------------------------------------------------------------- /src/tools/eval_Divot_video_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/tools/eval_Divot_video_gen.py -------------------------------------------------------------------------------- /src/tools/eval_Divot_video_recon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/tools/eval_Divot_video_recon.py -------------------------------------------------------------------------------- /src/tools/merge_agent_lora_weight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/tools/merge_agent_lora_weight.py -------------------------------------------------------------------------------- /src/train/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/train/dist_utils.py -------------------------------------------------------------------------------- /src/train/optim.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/train/schedular.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/train/schedular.py -------------------------------------------------------------------------------- /src/train/train_Divot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/Divot/HEAD/src/train/train_Divot.py --------------------------------------------------------------------------------