renate.utils package

Submodules

• renate.utils.avalanche module
  • BaseAvalancheDataset
  • AvalancheDataset
  • AvalancheDatasetForBuffer
  • to_avalanche_dataset()
  • AvalancheBenchmarkWrapper
    • AvalancheBenchmarkWrapper.update_benchmark_properties()
    • AvalancheBenchmarkWrapper.state_dict()
    • AvalancheBenchmarkWrapper.load_state_dict()
  • replace_plugin()
  • remove_plugin()
  • plugin_by_class()
• renate.utils.config_spaces module
  • config_space()
• renate.utils.deepspeed module
  • ds_checkpoint_dir()
  • search_key()
  • convert_zero_checkpoint_to_fp32_state_dict()
  • convert_to_tensor()
  • recover_object_from_tensor()
• renate.utils.distributed_strategies module
  • create_strategy()
• renate.utils.file module
  • get_aws_region()
  • get_bucket()
  • is_s3_uri()
  • move_to_uri()
  • copy_to_uri()
  • maybe_download_from_s3()
  • download_folder_from_s3()
  • upload_folder_to_s3()
  • download_file_from_s3()
  • upload_file_to_s3()
  • delete_file_from_s3()
  • extract_file()
  • download_file()
  • download_and_unzip_file()
  • save_pandas_df_to_csv()
  • unlink_file_or_folder()
• renate.utils.hf_utils module
  • DataCollatorWithPaddingForWildTime
    • DataCollatorWithPaddingForWildTime.tokenizer
• renate.utils.misc module
  • int_or_str()
  • maybe_populate_mask_and_ignore_logits()
  • AdditionalTrainingMetrics
    • AdditionalTrainingMetrics.on_train_start()
    • AdditionalTrainingMetrics.on_train_epoch_end()
    • AdditionalTrainingMetrics.parameters_count()
• renate.utils.module module
  • evaluate_and_record_results()
  • get_model()
  • get_data_module()
  • get_loss_fn()
  • get_optimizer()
  • get_learning_rate_scheduler()
  • get_metrics()
  • get_and_prepare_data_module()
  • get_and_setup_data_module()
  • import_module()
• renate.utils.optimizer module
  • create_partial_optimizer()
• renate.utils.pytorch module
  • reinitialize_model_parameters()
  • get_generator()
  • randomly_split_data()
  • move_tensors_to_device()
  • get_length_nested_tensors()
  • cat_nested_tensors()
  • unique_classes()
  • complementary_indices()
  • ConcatRandomSampler
• renate.utils.syne_tune module
  • TuningLoggerCallback
    • TuningLoggerCallback.on_trial_result()
  • TrainingLoggerCallback
    • TrainingLoggerCallback.on_trial_result()
  • redirect_to_tmp()
  • config_space_to_dict()
  • config_space_from_dict()
  • best_hyperparameters()
  • is_syne_tune_config_space()