错误:无法识别的参数:-colab中的pytorch代码

时间:2019-04-22 10:45:18

标签: pytorch argparse google-colaboratory

我试图在Google colab 上运行代码。但是当我尝试调用此函数时,我收到此消息(error: unrecognized arguments

def parse_opts():
 parser = argparse.ArgumentParser()
 parser.add_argument(
     '--root_path',
     default='/root/data/ActivityNet',
     type=str,
     help='Root directory path of data')
  parser.add_argument(
     '--video_path',
     default='video_kinetics_jpg',
     type=str,
     help='Directory path of Videos')
 args = parser.parse_args()

return args

但是这失败了,我得到了这个错误

  

tester_video.py:错误:无法识别的参数:   cifar_comp_20_200_0.01_0.1 20 10 0.01 0.1

我尝试使用 Easydict ,但它似乎不起作用 认为

2 个答案:

答案 0 :(得分:0)

我解决了这个问题。您需要使用easydict而不是argparse。 插入上面的代码,您可以这样编写:

args = easydict.EasyDict(
{
    "root_path": '/root/data/ActivityNet',
    "video_path": 'video_kinetics_jpg',
    "annotation_path": 'kinetics.json',
    "result_path": 'results',
    "dataset": 'kinetics',
    "n_classes": 400,
    "n_finetune_classes": 400,
    "sample_size": 64,
    "sample_duration": 32,
    "initial_scale": 1.0,
    "n_scales": 5,
    "scale_step": 0.84089641525,
    "train_crop": 'corner',
    "learning_rate": 0.1 )}

答案 1 :(得分:0)

问题只出现在 Jupyter notebook/lab/colab 中。

改变

args = parser.parse_args()

args = parser.parse_args(args=[])

它应该修复它。