output
The base output class.
It implements the base output processing class, which will handle the model output saving and loading.
Attributes:
Name | Type | Description |
---|---|---|
name |
str, default = 'base_output'
|
Name of the output class object. |
Methods:
Name | Description |
---|---|
__init__ |
The initializatoutputn method of the base output class. |
save |
The output saving method of the output class. This method is declared to be a static method. |
load |
The output laoding method of the output class. This method is declared to be a static method. |
Source code in tinybig/output/base_output.py
16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 |
|
__init__(name='base_output', *args, **kwargs)
The base output class initializatoutputn method.
It initializes the base output class object for handling the output saving and loading.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
name
|
Name of the base output class object. |
'base_output'
|
Source code in tinybig/output/base_output.py
load(cache_dir='./result', output_file='output', *args, **kwargs)
staticmethod
The output result loading method.
It loads the output results from a binary file via the pick.load.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
cache_dir
|
Directory of the output result file. |
'./result'
|
|
output_file
|
Name of file of the output result. |
'output'
|
Returns:
Type | Description |
---|---|
None
|
The loaded output result. |
Source code in tinybig/output/base_output.py
save(result, cache_dir='./result', output_file='output', *args, **kwargs)
staticmethod
The output saving method.
It saves the provided outputs to a binary file via pick.dump.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
result
|
The output results to be saved. |
required | |
cache_dir
|
Directory to save the output result. |
'./result'
|
|
output_file
|
Name of file to save the output result. |
'output'
|
Returns:
Type | Description |
---|---|
None
|
This output result saving method doesn't have return values. |