hookeai.miscellaneous.pruning.pruning_dataset.perform_model_standard_training

perform_model_standard_training(train_dataset_file_path, model_directory, val_dataset_file_path=None, device_type='cpu', is_verbose=False)

Perform standard training of RNN-based model.

Parameters:

• train_dataset_file_path (str) – Training data set file path.

• model_directory (str) – Directory where model is stored.

• val_dataset_file_path (str, default=None) – Validation data set file path.

• device_type ({'cpu', 'cuda'}, default='cpu') – Type of device on which torch.Tensor is allocated.

• is_verbose (bool, default=False) – If True, enable verbose output.