aak / config.json
kiyoonkim's picture
Upload config.json with huggingface_hub
6755909
{"exp_name":"aak","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,"train_jitter_min":224,"train_jitter_max":336,"train_horizontal_flip":true,"test_scale":256,"test_num_spatial_crops":1,"crop_size":224,"mean":[0.485,0.456,0.406],"std":[0.229,0.224,0.225],"normalise":true,"bgr":false},"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.sgd.SGD","params":"???","lr":0.00001,"momentum":0.9,"dampening":0,"weight_decay":0.0005,"nesterov":false,"maximize":false,"foreach":null,"differentiable":false},"scheduler":{"_target_":"torch.optim.lr_scheduler.MultiStepLR","optimizer":"???","milestones":[5000,8000],"gamma":0.3,"last_epoch":-1,"verbose":false},"learner":{"_target_":"mlproject.boilerplate.Learner","experiment_name":"aak","experiment_dir":"/disk/scratch2/experiments/aak/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":"aak","repo_owner":"kiyoonkim"}},"wandb_args":{"_target_":"wandb.sdk.wandb_init.init","job_type":null,"dir":"/disk/scratch2/experiments/aak","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/aak","repo_path":"kiyoonkim/aak","hf_repo_dir":"/disk/scratch2/experiments/aak/repo","code_dir":"/home/kiyoon/project/mlproject"}