config.json

{"exp_name":"aag","model":{"_target_":"mlproject.models.tsm_resnet50.build_tsm_model","num_classes":51,"num_frames":8,"pretrained":"kinetics400"},"dataset":{"_target_":"mlproject.datasets.config.DatasetConfig","num_classes":51,"class_keys":null},"torch_dataset":{"_target_":"mlproject.torch_datasets.gulp_sparsesample.build_gulp_dataset","dataset_name":"hmdb51-gulprgb","data_dir":"/disk/scratch_fast1/datasets","sets_to_include":null},"dataloader":{"_target_":"torch.utils.data.dataloader.DataLoader","dataset":null,"batch_size":8,"shuffle":true,"sampler":null,"batch_sampler":null,"num_workers":4,"collate_fn":null,"pin_memory":true,"drop_last":false,"timeout":0.0,"worker_init_fn":null,"multiprocessing_context":null,"generator":null,"prefetch_factor":null,"persistent_workers":false,"pin_memory_device":""},"optimizer":{"_target_":"torch.optim.adamw.AdamW","params":"???","lr":0.00001,"betas":[0.9,0.999],"eps":1e-8,"weight_decay":0.00001,"amsgrad":false,"maximize":false,"foreach":null,"capturable":false,"differentiable":false,"fused":null},"scheduler":{"_target_":"torch.optim.lr_scheduler.MultiStepLR","optimizer":"???","milestones":[2000,4000,6000,8000],"gamma":0.1,"last_epoch":-1,"verbose":false},"learner":{"_target_":"mlproject.boilerplate.Learner","experiment_name":"aag","experiment_dir":"/disk/scratch2/experiments/aag/repo","model":null,"resume":false,"evaluate_every_n_steps":500,"evaluate_every_n_epochs":null,"checkpoint_every_n_steps":500,"checkpoint_after_validation":true,"train_iters":10000,"train_epochs":null,"train_dataloader":null,"limit_train_iters":null,"val_dataloaders":null,"limit_val_iters":null,"test_dataloaders":null,"trainers":null,"evaluators":null,"callbacks":null,"print_model_parameters":false},"callbacks":{"hf_uploader":{"_target_":"mlproject.callbacks.UploadCheckpointsToHuggingFace","repo_name":"aag","repo_owner":"kiyoonkim"}},"wandb_args":{"_target_":"wandb.sdk.wandb_init.init","job_type":null,"dir":"/disk/scratch2/experiments/aag","config":null,"project":"simple-tutorial","entity":null,"reinit":null,"tags":null,"group":null,"name":null,"notes":null,"magic":null,"config_exclude_keys":null,"config_include_keys":null,"anonymous":null,"mode":null,"allow_val_change":null,"resume":"allow","force":null,"tensorboard":null,"sync_tensorboard":null,"monitor_gym":null,"save_code":true,"id":null,"settings":null},"input_num_frames":8,"hf_username":"kiyoonkim","seed":42,"freeze_backbone":false,"resume":false,"resume_from_checkpoint":null,"print_config":false,"train_batch_size":8,"eval_batch_size":8,"num_workers":4,"train":true,"test":false,"download_latest":true,"download_checkpoint_with_name":null,"root_experiment_dir":"/disk/scratch2/experiments","data_dir":"/disk/scratch_fast1/datasets","current_experiment_dir":"/disk/scratch2/experiments/aag","repo_path":"kiyoonkim/aag","hf_repo_dir":"/disk/scratch2/experiments/aag/repo","code_dir":"/home/kiyoon/project/mlproject"}