Procházet zdrojové kódy

将GPU配置从是否开启改为数字配置

Prevalence před 7 roky
rodič
revize
586e91e0f2

+ 12 - 3
src/api/dlwork.js

@@ -44,7 +44,7 @@ export const getStudentDLProjectDetail = (dlworkId, userId) => {
  * @param type
  * @param name
  * @param dltype 机器学习框架类型,0为Tensorflow,1为Pytorch
- * @param useGPU 是否使用GPU
+ * @param numberOfGPU 是否使用GPU
  * @param description
  * @param startAt
  * @param endAt
@@ -53,7 +53,16 @@ export const getStudentDLProjectDetail = (dlworkId, userId) => {
  * @returns {Promise<*>}
  */
 export const postDLAssignment = (
-  { type, name, dltype, useGPU, description, startAt, endAt, problem },
+  {
+    type,
+    name,
+    dltype,
+    useGPU: numberOfGPU,
+    description,
+    startAt,
+    endAt,
+    problem
+  },
   courseId
 ) => {
   return request(`${DL_WORK_MODULE}/course/${courseId}/dlwork`, {
@@ -62,7 +71,7 @@ export const postDLAssignment = (
       type: type || null,
       name: name || null,
       dltype: dltype || 0,
-      useGPU: useGPU || false,
+      numberOfGPU: numberOfGPU || 0,
       description: description || null,
       startAt: startAt || null,
       endAt: endAt || null,

+ 8 - 7
src/components/AssignmentCRUD/CreateAssignment/index.vue

@@ -44,9 +44,11 @@
               style="flex: 1"
               >Pytorch
             </b-radio>
-            <b-checkbox value="false" v-model="$v.params.useGPU"
-              >是否使用GPU</b-checkbox
-            >
+            <b-input
+              type="number"
+              placeholder="使用GPU数量"
+              v-model="$v.params.numberOfGPU"
+            ></b-input>
           </div>
 
           <b-field
@@ -210,10 +212,9 @@ import { difficultyClass } from "@/data/questionDifficulty";
 import BInput from "buefy/src/components/input/Input";
 import BField from "buefy/src/components/field/Field";
 import BRadio from "buefy/src/components/radio/Radio";
-import BCheckbox from "buefy/src/components/checkbox/Checkbox";
 
 export default {
-  components: { BCheckbox, BRadio, BField, BInput },
+  components: { BRadio, BField, BInput },
   props: {
     courseId: {
       type: Number
@@ -239,7 +240,7 @@ export default {
       params: {
         name: null,
         dltype: 0,
-        useGPU: false,
+        numberOfGPU: 1,
         description: null,
         startAt: null,
         endAt: null,
@@ -410,7 +411,7 @@ export default {
       dltype: {
         required
       },
-      useGPU: {
+      numberOfGPU: {
         required
       },
       description: {