naive_pytorch_lib-master
setup.py
915B
.gitignore
2KB
cyy_naive_pytorch_lib
models
densenet.py
7KB
lenet.py
2KB
senet
se_inception.py
1KB
se_resnet.py
9KB
se_module.py
590B
se_resnet_group_norm.py
9KB
tensor.py
373B
dataset_collection.py
9KB
inference.py
9KB
test
test_hyper_parameter.py
175B
test_dataset.py
837B
__init__.py
0B
test_training.py
604B
data_structure
test_cuda_process_pool.py
757B
test_synced_tensor_dict_util.py
694B
test_cuda_process_task_queue.py
383B
algorithm
test_hessian_vector_product.py
3KB
test_conjugate_gradient.py
307B
test_per_sample_gradient.py
1KB
quantization
test_training.py
593B
hyper_parameter.py
8KB
dataset.py
13KB
configuration.py
702B
basic_trainer.py
9KB
ml_types.py
601B
model_with_loss.py
3KB
arg_parse.py
6KB
__init__.py
86B
trainer.py
6KB
model_executor.py
4KB
visualization.py
3KB
device.py
912B
model_factory.py
2KB
reproducible_env.py
3KB
gradient.py
571B
data_structure
synced_tensor_dict_util.py
2KB
__init__.py
0B
cuda_process_task_queue.py
616B
cuda_process_pool.py
368B
datasets
webank_street_dataset.py
2KB
__init__.py
0B
model_util.py
8KB
algorithm
conjugate_gradient.py
1KB
normalization.py
456B
per_sample_gradient.py
3KB
prune.py
3KB
steepest_descent.py
1KB
lr_finder.py
3KB
shapely_value
shapely_value.py
786B
shapely_value_factory.py
638B
quantization
trainer.py
4KB
__init__.py
0B
inverse_hessian_vector_product.py
3KB
hessian_vector_product.py
4KB
influence_function
classic_influence_function.py
1KB
hyper_gradient_analyzer.py
5KB
args.py
2KB
hyper_gradient_trainer.py
21KB
README.md
20B
暂无评论