Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/LiyuanLucasLiu/Torch-Scope
/ functions
Functions
36 in github.com/LiyuanLucasLiu/Torch-Scope
⨍
Functions
36
◇
Types & classes
5
↓ 13 callers
Method
format
(self, record)
torch_scope/wrapper.py:55
↓ 2 callers
Method
add_metric
Add metric value for the current expriments to the spreadsheet. Parameters ---------- metric_name: ``str``, required
torch_scope/sheet_writer.py:100
↓ 2 callers
Method
get_bytes
Convert other memory size to bytes Parameters ---------- size: ``str``, required. The numeric part of th
torch_scope/wrapper.py:493
↓ 2 callers
Method
restore_checkpoint
Restore checkpoint. Parameters ---------- folder_path: ``str``, required. Path to the checkpoint file
torch_scope/wrapper.py:147
↓ 2 callers
Method
save_config
save the config file.
torch_scope/sheet_writer.py:82
↓ 1 callers
Method
__init__
(self, path: str, name: str = None, seed: int = None,
torch_scope/wrapper.py:328
↓ 1 callers
Method
add_subparser
(self, name, parser)
torch_scope/commands.py:16
↓ 1 callers
Function
cached_url
Cached a file to the cache folder. Parameters ---------- url: ``str``, required. The url or the path.
torch_scope/file_manager.py:14
↓ 1 callers
Function
formatter_message
(message, use_color = True)
torch_scope/wrapper.py:40
↓ 1 callers
Method
nvidia_memory_map
Get the current GPU memory usage. Based on https://discuss.pytorch.org/t/access-gpu-memory-usage-in-pytorch/3192/4 Parameter
torch_scope/wrapper.py:473
↓ 1 callers
Function
read_readme
()
setup.py:6
↓ 1 callers
Method
restore_configue
Restore the config dict. Parameters ---------- name: ``str``, optional, (default = "config.json"). Name
torch_scope/wrapper.py:420
↓ 1 callers
Function
run
()
torch_scope/commands.py:33
Method
__init__
(self, msg, use_color = True)
torch_scope/wrapper.py:51
Method
__init__
(self, spread_sheet_name, path_for_worksheet_name, row_name, credential_path = None)
torch_scope/sheet_writer.py:27
Method
add_description
Add description for the experiment to the spreadsheet. Parameters ---------- description: ``str``, required.
torch_scope/wrapper.py:616
Method
add_description
Add descriptions for the current expriments to the spreadsheet. Parameters ---------- description: ``str``, required
torch_scope/sheet_writer.py:89
Method
add_loss_vs_batch
Add loss to the ``loss_tracking`` section in the tensorboard. Parameters ---------- kv_dict: ``dict``, required.
torch_scope/wrapper.py:633
Method
add_model_parameter_histograms
Add parameter histogram in the tensorboard. Parameters ---------- model: ``torch.nn.Module``, required.
torch_scope/wrapper.py:721
Method
add_model_parameter_stats
Add parameter stats to the ``parameter_*`` sections in the tensorboard. Parameters ---------- model: ``torch.nn.Modu
torch_scope/wrapper.py:663
Method
add_model_update_stats
Add parameter update stats to the ``parameter_gradient_update`` sections in the tensorboard. Parameters ---------- m
torch_scope/wrapper.py:698
Method
auto_device
Automatically choose the gpu (would return the gpu index with minimal used gpu memory by default). Parameters __________
torch_scope/wrapper.py:240
Method
auto_device
Automatically choose the gpu (would return the gpu index with minimal used gpu memory by default). Parameters __________
torch_scope/wrapper.py:506
Function
checkout_checkpoint
(args)
torch_scope/commands.py:23
Method
close
Close the tensorboard writer and the logger.
torch_scope/wrapper.py:610
Method
confirm_an_empty_path
Check whether a folder is an empty folder (not-exist). Parameters __________ path: ``str``, required. Pa
torch_scope/wrapper.py:524
Method
get_bytes
Convert other memory size to bytes Parameters ---------- size: ``str``, required. The numeric part of th
torch_scope/wrapper.py:210
Method
get_writer
Return the tensorboard writer.
torch_scope/wrapper.py:604
Method
nvidia_memory_map
Get the current GPU memory usage. Based on https://discuss.pytorch.org/t/access-gpu-memory-usage-in-pytorch/3192/4 Parameter
torch_scope/wrapper.py:164
Method
restore_best_checkpoint
Restore the best checkpoint. Parameters ---------- folder_path: ``str``, required. Path to the folder co
torch_scope/wrapper.py:130
Method
restore_best_checkpoint
Restore the best checkpoint. Parameters ---------- folder_path: ``str``, optional, (default = None). Pat
torch_scope/wrapper.py:454
Method
restore_configue
Restore the config dict. Parameters ---------- path: ``str``, required. The path toward the folder.
torch_scope/wrapper.py:89
Method
restore_latest_checkpoint
Restore the latest checkpoint. Parameters ---------- folder_path: ``str``, required. Path to the folder
torch_scope/wrapper.py:106
Method
restore_latest_checkpoint
Restore the latest checkpoint. Parameters ---------- folder_path: ``str``, optional, (default = None). P
torch_scope/wrapper.py:435
Method
save_checkpoint
Save checkpoint under the path. Parameters ---------- model: ``torch.nn.Module``, required. The model to
torch_scope/wrapper.py:567
Method
save_configue
Save config dict to the ``config.json`` under the path. Parameters ---------- config: ``dict``, required.
torch_scope/wrapper.py:549