AttributeError: module 'torch._C' has no attribute 'set_grad_enabled'

[jim79]

Hi, When I run the baseline.py the following attribute error message comes up. Can you kindly help.

Thank you JIM

/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py:566: UserWarning: This DataLoader will create 4 worker processes in total. Our suggested max number of worker in current system is 2, which is smaller than what this DataLoader is going to create. Please be aware that excessive worker creation might get DataLoader running slow or even freeze, lower the worker number to avoid potential slowness/freeze if necessary.
  cpuset_checked))
Traceback (most recent call last):
  File "baseline.py", line 225, in <module>
    main()
  File "baseline.py", line 221, in main
    work.train()
  File "/usr/local/lib/python3.7/dist-packages/flerken/framework/pytorchframework.py", line 74, in inner
    return func(*args, **kwargs)
  File "/usr/local/lib/python3.7/dist-packages/flerken/framework/pytorchframework.py", line 63, in inner
    return func(*args, **kwargs)
  File "baseline.py", line 96, in train
    with train(self):
  File "/usr/local/lib/python3.7/dist-packages/flerken/framework/pytorchframework.py", line 389, in __enter__
    torch._C.set_grad_enabled(True)
AttributeError: module 'torch._C' has no attribute 'set_grad_enabled'