(args)
| 41 | return parser.parse_args() |
| 42 | |
| 43 | def prepare_model(args): |
| 44 | if args.method == 'protonet': |
| 45 | from methods.protonet import ProtoNet |
| 46 | model = ProtoNet(args) |
| 47 | elif args.method == 'relationnet': |
| 48 | from methods.relationnet import RelationNet |
| 49 | model = RelationNet(args) |
| 50 | return model |
| 51 | |
| 52 | def prepare_optimizer(args, model): |
| 53 | if args.backbone == 'Conv4': |