9 lines
142 B
Python
9 lines
142 B
Python
import torch
|
|
def main():
|
|
print(torch.__version__)
|
|
print("Hello from attention-from-scratch!")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|