FGiuliari/Trajectory-Transformer

Typos in example

manybodycpa opened this issue · 1 comments

The example for the individual transformer contains two typos:

CUDA_VISIBLE_DEVICES

should read

CUDA_VISIBLE_DEVICES=0

and

--max_epoch

requires an argument.

Thank you for the notification, I've updated the readme.