search

mx-mark / VideoTransformer-pytorch

PyTorch implementation of a collections of scalable Video Transformer Benchmarks.
281 stars 35 forks source link

can i have a dockerimage? #2

Closed SangMyeongWoh closed 2 years ago

SangMyeongWoh commented 2 years ago

core dumped error occur while run script.

can i have a docker images?

mx-mark commented 2 years ago

Would you mind to comment out the code 'os.environ['NCCL_SOCKET_IFNAME'] = args.nccl_ifname #'lan1'' in the line 107 of model_pretrain,py, and to see if it works?

SangMyeongWoh commented 2 years ago

Would you mind to comment out the code 'os.environ['NCCL_SOCKET_IFNAME'] = args.nccl_ifname #'lan1'' in the line 107 of model_pretrain,py, and to see if it works?

nothing changed T^T

i used model_inference.py and it happend while importing

model_inference.py, line 8 import data_transform as T

mx-mark commented 2 years ago

@SangMyeongWoh fine, we will build a notebook to make the model inference more convenient

SangMyeongWoh commented 2 years ago

Owner

thanks for your effort!