벼랑끝 바이오 매드 컴퓨터 사이언티스트

brief solution. 1. Whenever you train your model. You must check if your model and input data both are on the same device. (cpu or cuda) 2. And you must check the input data type and model parameters' data type are same Input type은 torch.FloatTensor인데 weight type(model parameter)는 torch.DoubleTensor라서 나타나는 에러라고 생각된다. for epoch in range(train_epoch+1): model.train() for i, (v_d, v_p, label) in en..