|
| | train.print_header (text) |
| | train.format_time (seconds) |
| | train.get_device () |
| | train.init_wandb (args, phase_name) |
| | train.build_loader (dataset, batch_size, shuffle, workers, custom_sampler=None) |
| | train.parse_data_roots (args) |
| | train.resolve_data_splits (data_roots) |
| | train.build_phase14_datasets (data_roots, n_globals) |
| | train.phase_eta_min (phase) |
| | train.build_scheduler (optimizer, total_steps, warmup_steps, eta_min) |
| | train.split_steps_for_epoch (total_steps, val_splits) |
| | train.arch_string (args) |
| | train.phase1_export_state_dict (model) |
| | train.save_model_checkpoint (path, model, phase, epoch, val_terms, args, extra=None, state_dict_override=None) |
| | train.empty_loss_terms () |
| | train.add_loss_terms (sums, total_t, wdl_t, n) |
| | train.finalize_loss_terms (sums, total_samples) |
| | train.format_loss_terms (tag, terms, color) |
| | train.to_planes_list (branches) |
| | train.top2_sparse_weights (base_weights) |
| | train.unpack_common_batch (batch, device) |
| | train.forward_backbone (model, planes_list, bypass, global_feats) |
| | train.train_phase1_steps (model, data_iter, num_steps, optimizer, scheduler, device) |
| | train.validate_phase1 (model, loader, device) |
| | train.copy_expert0_to_all (model) |
| | train.run_phase1 (model, args, wandb_run=None) |
| | train.train_expert_steps (model, expert_idx, data_iter, num_steps, optimizer, scheduler, device) |
| | train.validate_expert (model, expert_idx, loader, device) |
| | train.load_checkpoint_into_model (model, checkpoint_path, device) |
| | train.run_phase2 (model, args, wandb_run=None) |
| | train.train_phase4_steps (model, data_iter, num_steps, optimizer, scheduler, device) |
| | train.validate_phase4 (model, loader, device) |
| | train.run_phase4 (model, args, wandb_run=None) |
| | train.main () |
1.17.0