(args)
| 148 | |
| 149 | |
| 150 | def create_experiment_directory(args): |
| 151 | os.makedirs(args.results_dir, exist_ok=True) # Make results folder (holds all experiment subfolders) |
| 152 | experiment_index = len(glob(os.path.join(args.results_dir, "*"))) |
| 153 | model_string_name = args.model.replace("/", "-") # e.g., Latte-XL/2 --> Latte-XL-2 (for naming folders) |
| 154 | num_frame_string = f"F{args.num_frames}S{args.frame_interval}" |
| 155 | experiment_dir = os.path.join( # Create an experiment folder |
| 156 | args.results_dir, |
| 157 | f"{experiment_index:03d}-{model_string_name}-{num_frame_string}-{args.dataset}" |
| 158 | ) |
| 159 | experiment_dir = get_experiment_dir(experiment_dir, args) |
| 160 | checkpoint_dir = os.path.join(experiment_dir, "checkpoints") # Stores saved model checkpoints |
| 161 | os.makedirs(checkpoint_dir, exist_ok=True) |
| 162 | |
| 163 | return experiment_dir, checkpoint_dir |
| 164 | |
| 165 | |
| 166 | def main(args): |
no test coverage detected