<filename>main.py
import argparse
import os
import sys
import random

import numpy as np
import torch
import torchvision
from torch.nn import CrossEntropyLoss

if __name__ == '__main__':
    print(torch.__version__)
    print("Hello world")
<filename>main.py
import argparse
import os
import sys
import random

import numpy as np
import torch
import torchvision
from torch.nn import CrossEntropyLoss

if __name__ == '__main__':
    print(torch.__version__)
    print("Hello world")