diff --git a/train.py b/train.py index f6df6862009ae0f6bc812b307788c183eb55c329..2c3ba2fd84a5d4022b9d2a75e6ba4d23365ab6a5 100644 --- a/train.py +++ b/train.py @@ -24,7 +24,7 @@ from models import create_model from utils import parse_RAM_info, parse_nvidia_smi, visualizeForwardsNoGrad, printDict - +# some comments if __name__ == '__main__':