importpytorch_influence_functionsasptif# Supplied by the user:model=get_my_model()trainloader,testloader=get_my_dataloaders()ptif.init_logging()config=ptif.get_default_config()influences,harmful,helpful=ptif.calc_img_wise(config,model,trainloader,testloader)# do someting with influences/harmful/...
2 changes: 1 addition & 1 deletion 2 pytorch_influence_functions/calc_influence_function.py Original file line numberDiff line numberDiff line change @@ -458,7 +458,7 @@ def calc_img_wise(config, model, train_loader, test_loader): start_time = time.time() influence, harmful, helpful...
8 changes: 8 additions & 0 deletions 8 pytorch_influence_functions/calc_influence_function.py Original file line numberDiff line numberDiff line change @@ -365,6 +365,11 @@ def get_dataset_sample_ids_per_class(class_id, num_samples, test_loader,...
pytorch_influence_functions.egg-info/ pyvenv.cfg .vscode/ .idea/ 6 changes: 5 additions & 1 deletion 6 README.md Original file line numberDiff line numberDiff line change @@ -56,7 +56,11 @@ You can either install this package directly through pip: pip3 install --user pytorch-influe...