Use amp and ImportError

[zzzyzh]

When I set --use_amp to True, I get the following traceback:

Traceback (most recent call last):
File "train.py", line 16, in
from apex import amp
File "/home/miniconda3/envs/msa/lib/python3.8/site-packages/apex/init.py", line 13, in
from pyramid.session import UnencryptedCookieSessionFactoryConfig
ImportError: cannot import name 'UnencryptedCookieSessionFactoryConfig' from 'pyramid.session' (unknown location)

[Hk416mod2]

jeonsworld/ViT-pytorch#23